We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e28e007 commit 8b3d33aCopy full SHA for 8b3d33a
paddle/operators/sgd_op.h
@@ -30,7 +30,7 @@ class SGDOpKernel : public framework::OpKernel {
30
void Compute(const framework::ExecutionContext& ctx) const override {
31
auto param = ctx.Input<Tensor>("param");
32
auto grad = ctx.Input<Tensor>("grad");
33
- auto param_out = ctx.Output<Tensor>(0);
+ auto param_out = ctx.Output<Tensor>("param_out");
34
float lr = ctx.op_.GetAttr<float>("learning_rate");
35
36
param_out->mutable_data<T>(ctx.GetPlace());
0 commit comments