提交 6818fb1a 编写于 作者: W wawltor

update the code check for the detetion_map_op, test=develop

上级 362850b9
......@@ -68,12 +68,12 @@ class DetectionMAPOp : public framework::OperatorWithKernel {
ctx->HasInput("TruePos"),
platform::errors::InvalidArgument(
"Input(TruePos) of DetectionMAPOp should not be null when "
"Input(TruePos) is not null."));
"Input(PosCount) is not null."));
PADDLE_ENFORCE(
ctx->HasInput("FalsePos"),
platform::errors::InvalidArgument(
"Input(FalsePos) of DetectionMAPOp should not be null when "
"Input(FalsePos) is not null."));
"Input(PosCount) is not null."));
}
ctx->SetOutputDim("MAP", framework::make_ddim({1}));
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册