wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-27 13:12:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixes #250 (#253)

Browse Source
This commit is contained in:
William Falcon authored and GitHub committed 2019-09-26 09:13:00 -04:00
1 parent b0a0a47a0b
commit 3337c0237b
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
docs/LightningModule/RequiredTrainerInterface.md
+1 -1
View File
@@ -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