From 9aa9a1a796640ebe0d0b917d9c3f5ee0c39a1ab5 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 17 Aug 2019 11:11:07 -0400 Subject: [PATCH] Update lightning_module_template.py --- examples/new_project_templates/lightning_module_template.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/new_project_templates/lightning_module_template.py b/examples/new_project_templates/lightning_module_template.py index 94e3407d..adc81f44 100644 --- a/examples/new_project_templates/lightning_module_template.py +++ b/examples/new_project_templates/lightning_module_template.py @@ -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)