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

updated optimizer_step docs

Browse Source
This commit is contained in:
William Falcon
2019-08-13 11:51:31 -04:00
parent 89c4c260ad
commit 7898d0c02a
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/LightningModule/RequiredTrainerInterface.md
+4 -1
View File
@@ -200,7 +200,10 @@ def configure_optimizers(self):
### validation_step ### validation_step
``` {.python} ``` {.python}
def validation_step(self, data_batch, batch_nb, dataloader_i) def validation_step(self, data_batch, batch_nb)
# if have multiple val dataloaders:
def validation_step(self, data_batch, batch_nb, dataloader_idx)
``` ```
**OPTIONAL** **OPTIONAL**
If you don't need to validate you don't need to implement this method. If you don't need to validate you don't need to implement this method.