p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!3060 modify init value of mean square in rmsprop optimizer 

Merge pull request !3060 from wangnan39/modify_mean_square_init_value_in_rmsprop
This commit is contained in:
mindspore-ci-bot 2020-07-14 20:50:59 +08:00 committed by Gitee
parent c99cc0dfa1 046e369e36
commit 1b9f0fc208
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/optim/rmsprop.py
View File

@ -171,7 +171,7 @@ class RMSProp(Optimizer):
self.opt = P.ApplyRMSProp(use_locking)
self.momentum = momentum
self.ms = self.parameters.clone(prefix="mean_square", init='zeros')
self.ms = self.parameters.clone(prefix="mean_square", init='ones')
self.moment = self.parameters.clone(prefix="moment", init='zeros')
self.hyper_map = C.HyperMap()
self.epsilon = epsilon