Update RequiredTrainerInterface.md

This commit is contained in:
William Falcon
2019-07-26 08:24:56 -04:00
committed by GitHub
parent ba25161dcc
commit 5ba0a8fe4c
@@ -234,7 +234,7 @@ Tuple - List of optimizers and list of schedulers
``` {.python}
# most cases
def configure_optimizers(self):
opt = Adam(self.model.parameters(), lr=0.01)
opt = Adam(self.parameters(), lr=0.01)
return [opt], []
# gan example, with scheduler for discriminator