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

Merge branch 'hparams_from_checkpoint' of https://github.com/neggert/pytorch-lightning

Browse Source
This commit is contained in:
williamFalcon
2019-10-23 01:32:21 -07:00
parent f7dda5080b 3a2466258d
commit 3fcce57e6f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pl_examples/basic_examples/lightning_module_template.py
+1 -1
View File
@@ -158,7 +158,7 @@ class LightningTemplateModel(LightningModule):
val_loss = output['val_loss']
# reduce manually when using dp
if self.trainer.use_dp:
if self.trainer.use_dp or self.trainer.use_ddp2:
val_loss = torch.mean(val_loss)
val_loss_mean += val_loss