From 0e6462d6c671414918817090a781b5e8cf6358d1 Mon Sep 17 00:00:00 2001 From: sneaxiy <32832641+sneaxiy@users.noreply.github.com> Date: Tue, 14 Jun 2022 16:51:03 +0800 Subject: [PATCH] fix update loss scaling (#43487) --- paddle/fluid/operators/amp/update_loss_scaling_op.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/operators/amp/update_loss_scaling_op.cc b/paddle/fluid/operators/amp/update_loss_scaling_op.cc index baf742b0b40..26fa1c91316 100644 --- a/paddle/fluid/operators/amp/update_loss_scaling_op.cc +++ b/paddle/fluid/operators/amp/update_loss_scaling_op.cc @@ -107,9 +107,9 @@ class UpdateLossScalingOpMaker : public framework::OpProtoAndCheckerMaker { AddOutput("LossScaling", "(Tensor) 1-dim tensor, updated loss scaling."); AddOutput("OutGoodSteps", "(Tensor) 1-dim tensor, pdated good steps."); AddOutput("OutBadSteps", "(Tensor) 1-dim tensor, updated bad steps."); - AddOutput("StopUpdate", - "(Tensor) 1-dim tensor. Stop updating loss scaling, and just " - "zero inputs. It has higher priority than Attr(stop_update).") + AddInput("StopUpdate", + "(Tensor) 1-dim tensor. Stop updating loss scaling, and just " + "zero inputs. It has higher priority than Attr(stop_update).") .AsDispensable(); AddAttr("incr_every_n_steps", "A value represents increasing loss scaling every n " -- GitLab