From 73a836f286d7cae466deb240da462f27ce580f08 Mon Sep 17 00:00:00 2001 From: zhaojichen Date: Wed, 3 Jun 2020 23:04:50 -0400 Subject: [PATCH] fix applyrmsprop --- mindspore/ccsrc/transform/op_declare.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ccsrc/transform/op_declare.cc b/mindspore/ccsrc/transform/op_declare.cc index 735c9aac09..ee59d56003 100644 --- a/mindspore/ccsrc/transform/op_declare.cc +++ b/mindspore/ccsrc/transform/op_declare.cc @@ -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())}, {7, ATTR_DESC(momentum, AnyTraits())}, {8, ATTR_DESC(epsilon, AnyTraits())}};