training batch clean up

Lightning-AI · Jun 8, 2020 · 3676b6c · 3676b6c
1 parent cd9da57
commit 3676b6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -881,7 +881,7 @@ def fit(
                 mp.spawn(self.ddp_train, nprocs=self.num_processes, args=(model,))
 
             elif self.distributed_backend == 'ddp_spawn':
-                model.share_memory()
+                # model.share_memory()
 
                 # spin up peers
                 mp.spawn(self.ddp_train, nprocs=self.num_processes, args=(model, ))