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

Update lm_test_module_mixins.py

Browse Source
This commit is contained in:
William Falcon
2019-10-23 04:29:13 -04:00
committed by GitHub
parent 7092b6cb94
commit c5c03c87db
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/testing/lm_test_module_mixins.py
+2 -2
View File
@@ -80,13 +80,13 @@ class LightningValidationMixin(LightningValidationStepMixin):
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
# reduce manually when using dp
val_acc = output['val_acc']
if self.trainer.use_dp:
if self.trainer.use_dp or self.trainer.use_ddp2:
val_acc = torch.mean(val_acc)
val_acc_mean += val_acc