From 3337c0237b004a8b0df77eea40d77e50a8e0e558 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 26 Sep 2019 09:13:00 -0400 Subject: [PATCH] Fixes #250 (#253) --- docs/LightningModule/RequiredTrainerInterface.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/LightningModule/RequiredTrainerInterface.md b/docs/LightningModule/RequiredTrainerInterface.md index d8ccae94f3..6308d5696c 100644 --- a/docs/LightningModule/RequiredTrainerInterface.md +++ b/docs/LightningModule/RequiredTrainerInterface.md @@ -145,7 +145,7 @@ def training_step(self, batch, batch_nb): output = { 'loss': loss, # required - 'progress': {'training_loss': loss, 'batch_nb': batch_nb} # optional + 'progress': {'training_loss': loss} # optional (MUST ALL BE TENSORS) } # return a dict