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

updated optimizer_step docs

Browse Source
This commit is contained in:
William Falcon committed 2019-08-13 11:51:31 -04:00
1 parent 89c4c260ad
commit 7898d0c02a
1 file changed
+4 -1
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
``` {.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**
If you don't need to validate you don't need to implement this method.