added arg docs

2019-07-18 12:09:25 -04:00 · 2019-07-18 12:09:25 -04:00 · 8be7480f31
parent 751bc7c695
commit 8be7480f31
1 changed files with 2 additions and 0 deletions
--- 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: