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 lightning_module_template.py

Browse Source
This commit is contained in:
William Falcon
2019-08-17 11:11:07 -04:00
committed by GitHub
parent 48de39ed50
commit 9aa9a1a796
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/new_project_templates/lightning_module_template.py
+2 -2
View File
@@ -161,9 +161,9 @@ class LightningTemplateModel(LightningModule):
# reduce manually when using dp
val_acc = output['val_acc']
if self.trainer.use_dp:
val_acc_mean = torch.mean(val_acc)
val_acc = torch.mean(val_acc)
val_acc_mean += val_acc_mean
val_acc_mean += val_acc
val_loss_mean /= len(outputs)
val_acc_mean /= len(outputs)