From bc6c4dbb9fa06ade3bc6da36fbd6265cec163be4 Mon Sep 17 00:00:00 2001 From: Yuan Gao Date: Sun, 28 Jan 2018 16:00:55 +0800 Subject: [PATCH] Update box_coder_op.cc --- paddle/operators/box_coder_op.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/operators/box_coder_op.cc b/paddle/operators/box_coder_op.cc index 0cb20a4182e..7d0d28aabaa 100644 --- a/paddle/operators/box_coder_op.cc +++ b/paddle/operators/box_coder_op.cc @@ -32,15 +32,15 @@ class BoxCoderOp : public framework::OperatorWithKernel { auto target_box_dims = ctx->GetInputDim("TargetBox"); PADDLE_ENFORCE_EQ(prior_box_dims.size(), 2UL, - "The shape of PriorBox is [N, 4]"); + "The rank of Input of PriorBox must be 2"); PADDLE_ENFORCE_EQ(prior_box_dims[1], 4UL, "The shape of PriorBox is [N, 4]"); PADDLE_ENFORCE_EQ(prior_box_var_dims.size(), 2UL, - "The shape of PriorBoxVar is [N, 4]"); + "The rank of Input of PriorBoxVar must be 2"); PADDLE_ENFORCE_EQ(prior_box_var_dims[1], 4UL, "The shape of PriorBoxVar is [N, 4]"); PADDLE_ENFORCE_EQ(target_box_dims.size(), 2UL, - "The shape of TargetBox is [M, 4]"); + "The rank of Input of TargetBox must be 2"); PADDLE_ENFORCE_EQ(target_box_dims[1], 4UL, "The shape of TargetBox is [M, 4]"); -- GitLab