Skip to content

Commit 8b3d33a

Browse files
committed
fix-sgd
1 parent e28e007 commit 8b3d33a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/operators/sgd_op.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ class SGDOpKernel : public framework::OpKernel {
3030
void Compute(const framework::ExecutionContext& ctx) const override {
3131
auto param = ctx.Input<Tensor>("param");
3232
auto grad = ctx.Input<Tensor>("grad");
33-
auto param_out = ctx.Output<Tensor>(0);
33+
auto param_out = ctx.Output<Tensor>("param_out");
3434
float lr = ctx.op_.GetAttr<float>("learning_rate");
3535

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

0 commit comments

Comments
 (0)