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

update the code check for the detetion_map_op, test=develop

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