Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixes #250 (#253)

This commit is contained in:
William Falcon 2019-09-26 09:13:00 -04:00 committed by GitHub
parent b0a0a47a0b
commit 3337c0237b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/LightningModule/RequiredTrainerInterface.md
View File

@ -145,7 +145,7 @@ def training_step(self, batch, batch_nb):
output = { output = {
'loss': loss, # required 'loss': loss, # required
'progress': {'training_loss': loss, 'batch_nb': batch_nb} # optional 'progress': {'training_loss': loss} # optional (MUST ALL BE TENSORS)
} }
# return a dict # return a dict