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 #361 (#391)

Browse Source
This commit is contained in:
William Falcon authored and GitHub committed 2019-10-19 00:39:45 +02:00
1 parent 6111edaf82
commit c1bbc2158f
2 files changed
+36 -7
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

docs/LightningModule/RequiredTrainerInterface.md
+6 -1
View File
@@ -143,10 +143,15 @@ def training_step(self, batch, batch_nb):
out = self.forward(x)
loss = self.loss(out, x)
logger_logs = {'training_loss': loss} # optional (MUST ALL BE TENSORS)
# if using TestTubeLogger or TensorboardLogger you can nest scalars
logger_logs = {'losses': logger_logs} # optional (MUST ALL BE TENSORS)
output = {
'loss': loss, # required
'progress_bar': {'training_loss': loss}, # optional (MUST ALL BE TENSORS)
'log': {'training_loss': loss} # optional (MUST ALL BE TENSORS)
'log': logger_logs
}
# return a dict