wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-10 12:21:57 +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:54:19 -04:00
parent 7898d0c02a
commit 7e38f1f246
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
+3
View File
@@ -195,6 +195,9 @@ def configure_optimizers(self):
return [generator_opt, disriminator_opt], [discriminator_sched] return [generator_opt, disriminator_opt], [discriminator_sched]
``` ```
If you need to control how often those optimizers step or override the default .step() schedule, override
the [optimizer_step](https://williamfalcon.github.io/pytorch-lightning/Trainer/hooks/#optimizer_step) hook.
## Optional Methods ## Optional Methods
### validation_step ### validation_step