From 8be7480f31194ee9952f8d44d68ac296c0aa7039 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 18 Jul 2019 12:09:25 -0400 Subject: [PATCH] added arg docs --- examples/new_project_templates/lightning_module_template.py | 2 ++ 1 file changed, 2 insertions(+) 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: