From f01cb63234e2fa717fe953cb4b0787f04c45d8f1 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 18 Jul 2019 12:10:07 -0400 Subject: [PATCH] added arg docs --- examples/new_project_templates/lightning_module_template.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/new_project_templates/lightning_module_template.py b/examples/new_project_templates/lightning_module_template.py index 91fe8c3e03..a0c65af546 100644 --- a/examples/new_project_templates/lightning_module_template.py +++ b/examples/new_project_templates/lightning_module_template.py @@ -116,7 +116,7 @@ class LightningTemplateModel(LightningModule): :param outputs: list of individual outputs of each validation step :return: """ - return outputs.mean() + return torch.stack(outputs).mean() val_loss_mean = 0 val_acc_mean = 0