From 68ca577919a111e3ad49a9bf9b52af95cfa4c8c8 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 23 Apr 2020 15:03:39 -0400 Subject: [PATCH] why copy? (#1579) --- .gitignore | 5 +++++ pytorch_lightning/trainer/training_loop.py | 4 ++-- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/.gitignore b/.gitignore index 59fb3269ce..d5bea7f6d5 100644 --- a/.gitignore +++ b/.gitignore @@ -131,3 +131,8 @@ ENV/ .data/ datasets/ mnist/ + +# pl tests +ml-runs/ +*.zip +pytorch\ lightning \ No newline at end of file diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py index b41f159773..b6756d1e94 100644 --- a/pytorch_lightning/trainer/training_loop.py +++ b/pytorch_lightning/trainer/training_loop.py @@ -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)