diff --git a/examples/new_project_templates/lightning_module_template.py b/examples/new_project_templates/lightning_module_template.py index c3ecaff51c..91fe8c3e03 100644 --- a/examples/new_project_templates/lightning_module_template.py +++ b/examples/new_project_templates/lightning_module_template.py @@ -116,6 +116,8 @@ class LightningTemplateModel(LightningModule): :param outputs: list of individual outputs of each validation step :return: """ + return outputs.mean() + val_loss_mean = 0 val_acc_mean = 0 for output in outputs: