Copy batch for local forward

Lightning-AI · Nov 18, 2019 · 9142a1f · 9142a1f
1 parent e22dea2
commit 9142a1f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/train_loop_mixin.py b/pytorch_lightning/trainer/train_loop_mixin.py
@@ -290,7 +290,7 @@ def training_forward(self, batch, batch_nb, opt_idx, hiddens):
             gpu_id = 0
             if type(self.data_parallel_device_ids) is list:
                 gpu_id = self.data_parallel_device_ids[0]
-            batch = self.transfer_batch_to_gpu(batch, gpu_id)
+            batch = self.transfer_batch_to_gpu(batch.copy(), gpu_id)
             args[0] = batch
             output = self.model.training_step(*args)