diff --git a/paddle/fluid/operators/metrics/accuracy_op.h b/paddle/fluid/operators/metrics/accuracy_op.h old mode 100755 new mode 100644 index 803244dd48efc634bf5e654a35cb3dd572842882..94e5bf8257e67b9fd01aa9ae45a25d90963fef13 --- a/paddle/fluid/operators/metrics/accuracy_op.h +++ b/paddle/fluid/operators/metrics/accuracy_op.h @@ -50,7 +50,11 @@ class AccuracyKernel : public framework::OpKernel { int num_correct = 0; // assume inference is already the topk of the output for (size_t i = 0; i < num_samples; ++i) { - PADDLE_ENFORCE_GE(label_data[i], 0, "label must >= 0"); + PADDLE_ENFORCE_GE( + label_data[i], 0, + platform::errors::InvalidArgument( + "label of AccuracyOp must >= 0, But received label[%d] is %d", i, + label_data[i])); for (size_t j = 0; j < class_dim; ++j) { if (indices_data[i * class_dim + j] == label_data[i]) { ++num_correct;