fix-sgd

8 years ago · 8b3d33a055
parent e28e007373
commit 8b3d33a055
1 changed files with 1 additions and 1 deletions
--- a/paddle/operators/sgd_op.h
+++ b/paddle/operators/sgd_op.h
@ -30,7 +30,7 @@ class SGDOpKernel : public framework::OpKernel {
  void Compute(const framework::ExecutionContext& ctx) const override {
    auto param = ctx.Input<Tensor>("param");
    auto grad = ctx.Input<Tensor>("grad");
-    auto param_out = ctx.Output<Tensor>(0);
+    auto param_out = ctx.Output<Tensor>("param_out");
    float lr = ctx.op_.GetAttr<float>("learning_rate");

    param_out->mutable_data<T>(ctx.GetPlace());