Browse Source

fix applyrmsprop bug

tags/v0.5.0-beta
zhaojichen 5 years ago
parent
commit
b6a0ee9f6c
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/transform/op_declare.cc

+ 1
- 1
mindspore/ccsrc/transform/op_declare.cc View File

@@ -1226,7 +1226,7 @@ OUTPUT_MAP(Atan2) = {{0, OUTPUT_DESC(y)}};

// ApplyRMSPropD
INPUT_MAP(ApplyRMSPropD) = {
{1, INPUT_DESC(var)}, {2, INPUT_DESC(ms)}, {3, INPUT_DESC(mom)}, {4, INPUT_DESC(grad)}, {5, INPUT_DESC(lr)}};
{1, INPUT_DESC(var)}, {2, INPUT_DESC(ms)}, {3, INPUT_DESC(mom)}, {4, INPUT_DESC(lr)}, {5, INPUT_DESC(grad)}};
INPUT_ATTR_MAP(ApplyRMSPropD) = {{6, ATTR_DESC(rho, AnyTraits<float>())},
{7, ATTR_DESC(momentum, AnyTraits<float>())},
{8, ATTR_DESC(epsilon, AnyTraits<float>())}};


Loading…
Cancel
Save