learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-10-31 18:58:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

optim configs

Browse Source
This commit is contained in:
Varuna Jayasiri
2020-09-26 21:25:24 +05:30
parent 19bd8b8e67
commit b2b305ff4d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
labml_nn/gan/mnist.py
View File

@ -171,8 +171,8 @@ def _discriminator_optimizer(c: Configs):
opt_conf = OptimizerConfigs() opt_conf = OptimizerConfigs()
opt_conf.optimizer = 'Adam' opt_conf.optimizer = 'Adam'
opt_conf.parameters = c.discriminator.parameters() opt_conf.parameters = c.discriminator.parameters()
opt_conf.learning_rate = 2.5e-5 opt_conf.learning_rate = 2.5e-4
# opt_conf.betas = (0.5, 0.999) opt_conf.betas = (0.5, 0.999)
return opt_conf return opt_conf
@ -181,8 +181,8 @@ def _generator_optimizer(c: Configs):
opt_conf = OptimizerConfigs() opt_conf = OptimizerConfigs()
opt_conf.optimizer = 'Adam' opt_conf.optimizer = 'Adam'
opt_conf.parameters = c.generator.parameters() opt_conf.parameters = c.generator.parameters()
opt_conf.learning_rate = 2.5e-5 opt_conf.learning_rate = 2.5e-4
# opt_conf.betas = (0.5, 0.999) opt_conf.betas = (0.5, 0.999)
return opt_conf return opt_conf