diff --git a/examples/new_project_templates/lightning_module_template.py b/examples/new_project_templates/lightning_module_template.py index 5245e2cd..17b36573 100644 --- a/examples/new_project_templates/lightning_module_template.py +++ b/examples/new_project_templates/lightning_module_template.py @@ -84,8 +84,7 @@ class LightningTemplateModel(LightningModule): loss_val = self.loss(y, y_hat) output = OrderedDict({ - 'loss': loss_val, - 'tqdm_metrics': {} + 'loss': loss_val }) return output