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

why copy? (#1579)

Browse Source
This commit is contained in:
William Falcon
2020-04-23 15:03:39 -04:00
committed by GitHub
parent 29ebe92208
commit 68ca577919
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.gitignore
+5
View File
@@ -131,3 +131,8 @@ ENV/
.data/
datasets/
mnist/
# pl tests
ml-runs/
*.zip
pytorch\ lightning
pytorch_lightning/trainer/training_loop.py
+2 -2
View File
@@ -752,7 +752,7 @@ class TrainerTrainLoopMixin(ABC):
# Horovod
elif self.use_horovod and self.on_gpu:
batch = self.transfer_batch_to_gpu(copy.copy(batch), hvd.local_rank())
batch = self.transfer_batch_to_gpu(batch, hvd.local_rank())
args[0] = batch
output = self.model.training_step(*args)
@@ -771,7 +771,7 @@ class TrainerTrainLoopMixin(ABC):
# TPU support
elif self.use_tpu:
batch = self.transfer_batch_to_tpu(copy.copy(batch))
batch = self.transfer_batch_to_tpu(batch)
args[0] = batch
output = self.model.training_step(*args)