From fbc127279687d77d1d6e5aae62c4d762f3fb59d1 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 7 Oct 2019 20:08:54 -0400 Subject: [PATCH] Update lightning_module_template.py --- examples/basic_examples/lightning_module_template.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/basic_examples/lightning_module_template.py b/examples/basic_examples/lightning_module_template.py index a6d94595..c977fc14 100644 --- a/examples/basic_examples/lightning_module_template.py +++ b/examples/basic_examples/lightning_module_template.py @@ -102,7 +102,7 @@ class LightningTemplateModel(LightningModule): output = OrderedDict({ 'loss': loss_val, 'progress_bar': tqdm_dict, - 'loss': tqdm_dict + 'log': tqdm_dict }) # can also return just a scalar instead of a dict (return loss_val) @@ -171,7 +171,7 @@ class LightningTemplateModel(LightningModule): val_loss_mean /= len(outputs) val_acc_mean /= len(outputs) tqdm_dict = {'val_loss': val_loss_mean, 'val_acc': val_acc_mean} - result = {'progress_bar': tqdm_dict, 'logs': tqdm_dict} + result = {'progress_bar': tqdm_dict, 'log': tqdm_dict} return result # ---------------------