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

add aliases for optimizer names

Browse source
This commit is contained in:
Connor Olding 2019-03-22 12:57:47 +01:00
parent 86491ad841
commit b3b82ca4f0
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
onn/optimizer.py
View file

@ -585,3 +585,13 @@ class Padam(Adamlike):
power=1/8, debias=True, eps=1e-8): power=1/8, debias=True, eps=1e-8):
super().__init__(lr=lr, b1=b1, b2=b2, power=power, super().__init__(lr=lr, b1=b1, b2=b2, power=power,
debias=debias, runmax=True, yogi=False, eps=eps) debias=debias, runmax=True, yogi=False, eps=eps)
AMSGrad = AMSgrad
AdaDelta = Adadelta
AdaGrad = Adagrad
AdaMax = Adamax
NAdam = Nadam
PAdam = Padam
RMSProp = RMSprop
RMSPropCentered = RMSpropCentered