wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-29 13:30:57 +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:25:33 -04:00
1 parent 5f57792131
commit 970d1609e1
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
@@ -30,7 +30,7 @@ def reduce_distributed_output(output, nb_gpus):
return output
def dummy(gpu, a):
print(gpu)
print(a)
class Trainer(TrainerIO):