wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Loss keys (#387)

Browse Source 
* any key in logs or progress bar is a candidate for callback metric

* any key in logs or progress bar is a candidate for callback metric
This commit is contained in:
William Falcon
2019-10-18 15:28:13 +02:00
committed by GitHub
parent 0fac2d64cf
commit 3dfcef6994
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/LightningModule/RequiredTrainerInterface.md
+2 -4
View File
@@ -318,12 +318,10 @@ The ```dataset_idx``` corresponds to the order of datasets returned in ```val_da
``` {.python}
def validation_end(self, outputs)
```
If you didn't define a validation_step, this won't be called.
Called at the end of the validation loop with the outputs of validation_step.
If you didn't define a validation_step, this won't be called. Called at the end of the validation loop with the outputs of validation_step.
The outputs here are strictly for the progress bar. If you don't need to display anything, don't return anything.
Any keys present in 'log', 'progress_bar' or the rest of the dictionary are available for callbacks to access.
**Params**
| Param | description |