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

added single node distdataparallel

Browse Source
This commit is contained in:
William Falcon
2019-07-03 15:25:33 -04:00
parent 5f57792131
commit 970d1609e1
1 changed files with 1 additions and 1 deletions
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):