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)