wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

set dp as default backend

Browse Source
This commit is contained in:
William Falcon
2019-07-18 11:57:39 -04:00
parent b684bb55c5
commit 4085b3fa69
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
@@ -29,7 +29,7 @@ except ModuleNotFoundError:
def reduce_distributed_output(output, nb_gpus):
if nb_gpus <= 1:
if nb_gpus <= 1 or type(output) is torch.Tensor:
return output
for k, v in output.items():