Browse Source

modify init value of mean square in rmsprop optimizer

tags/v0.6.0-beta
wangnan39@huawei.com 5 years ago
parent
commit
046e369e36
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/optim/rmsprop.py

+ 1
- 1
mindspore/nn/optim/rmsprop.py View File

@@ -174,7 +174,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


Loading…
Cancel
Save