notwa/optim
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add RMSpropCentered to model from config

Browse Source
This commit is contained in:
Connor Olding 2017-09-26 23:12:40 +00:00
parent 9bb26b1ec5
commit 763246df98
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
onn.py
View File

@ -1034,6 +1034,10 @@ def optim_from_config(config):
d2 = config.optim_decay2 if 'optim_decay2' in config else 99.5
mu = np.exp(-1/d2)
optim = RMSprop(mu=mu)
elif config.optim == 'rmsc':
d2 = config.optim_decay2 if 'optim_decay2' in config else 9.5
mu = np.exp(-1/d2)
optim = RMSpropCentered(momentum=mu)
elif config.optim == 'sgd':
d1 = config.optim_decay1 if 'optim_decay1' in config else 0
clip = config.gradient_clip if 'gradient_clip' in config else 0.0