wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-27 13:12:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Copy batch for local forward (#532)

Browse Source
This commit is contained in:
Tullie Murrell authored and William Falcon committed 2019-11-23 04:04:40 -05:00
1 parent 55edf7c922
commit 48b797fdb0
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/train_loop_mixin.py
+1 -1
View File
@@ -290,7 +290,7 @@ class TrainerTrainLoopMixin(object):
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)