Fixes #250 (#253)

2019-09-26 09:13:00 -04:00 · 2019-09-26 09:13:00 -04:00 · 3337c0237b
parent b0a0a47a0b
commit 3337c0237b
1 changed files with 1 additions and 1 deletions
--- 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