From 4cbcc9b6da08dac482b69b5a4ee1296ecd13be9b Mon Sep 17 00:00:00 2001 From: Chengmo Date: Wed, 30 Dec 2020 16:45:19 +0800 Subject: [PATCH] fix momentum op register (#29941) * fix momentum op register --- paddle/fluid/operators/optimizers/momentum_op.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/paddle/fluid/operators/optimizers/momentum_op.cc b/paddle/fluid/operators/optimizers/momentum_op.cc index 1b01f5ebd87..b9a74c1bf71 100644 --- a/paddle/fluid/operators/optimizers/momentum_op.cc +++ b/paddle/fluid/operators/optimizers/momentum_op.cc @@ -118,6 +118,10 @@ REGISTER_OP_VERSION(momentum) Upgrade momentum add 2 attributes [regularization_method, regularization_coeff]. )ROC", paddle::framework::compatible::OpVersionDesc() + .NewInput("MasterParam", "FP32 master weight for AMP.") + .NewOutput("MasterParamOut", + "The updated FP32 master weight for AMP. " + "It shared memory with Input(MasterParam).") .NewAttr("regularization_method", "(string) regularization_method, right now only support " "l2decay or none", -- GitLab