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
+4 -1
View File
@@ -193,7 +193,10 @@ def configure_optimizers(self):
disriminator_opt = Adam(self.model_disc.parameters(), lr=0.02) disriminator_opt = Adam(self.model_disc.parameters(), lr=0.02)
discriminator_sched = CosineAnnealing(discriminator_opt, T_max=10) discriminator_sched = CosineAnnealing(discriminator_opt, T_max=10)
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