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

added single node distdataparallel

Browse Source
This commit is contained in:
William Falcon committed 2019-07-03 15:11:17 -04:00
1 parent c4aca832ba
commit 7ef6db49d3
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
+1 -1
View File
@@ -282,7 +282,7 @@ class Trainer(TrainerIO):
# when GPU is called, spawn off a single worker for each gpu
if self.on_gpu:
rank = 0
mp.spawn(self.__dp_train, nprocs=len(self.data_parallel_device_ids), args=(rank, model, self, ))
mp.spawn(self.__dp_train, nprocs=len(self.data_parallel_device_ids), args=(rank, model ))
else:
self.__run_pretrain_routine(model)