提交 d757c1cd 编写于 作者: D dangqingqing

Fix accuracy_op.

上级 48000a8f
...@@ -34,7 +34,7 @@ class AccuracyOp : public framework::OperatorWithKernel { ...@@ -34,7 +34,7 @@ class AccuracyOp : public framework::OperatorWithKernel {
PADDLE_ENFORCE_EQ(inference->dims()[0], label->dims()[0], PADDLE_ENFORCE_EQ(inference->dims()[0], label->dims()[0],
"inference size must be the same as label size"); "inference size must be the same as label size");
ctx.Output<Tensor>("Accuracy")->Resize({1}); ctx.Output<framework::LoDTensor>("Accuracy")->Resize({1});
} }
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册