diff --git a/mindspore/lite/src/executor.cc b/mindspore/lite/src/executor.cc index b8a816e5a9..0a4e4217c5 100644 --- a/mindspore/lite/src/executor.cc +++ b/mindspore/lite/src/executor.cc @@ -30,9 +30,9 @@ int Executor::CheckInputs(const std::vector &in_tensors) { return RET_ERROR; } auto shape = inTensor->shape(); - bool valid = all_of(shape.begin(), shape.end(), [](int i) { return i > 0; }); + bool valid = all_of(shape.begin(), shape.end(), [](int i) { return i >= 0; }); if (!valid) { - MS_LOG(ERROR) << "The shape of input tensor contains zero or negative dimension," + MS_LOG(ERROR) << "The shape of input tensor contains negative dimension," << "check the model and assign the input shape with method Resize()."; return RET_ERROR; } diff --git a/mindspore/lite/test/models_onnx_fp16.cfg b/mindspore/lite/test/models_onnx_fp16.cfg index fdcaf39c70..8e2d0c55cf 100644 --- a/mindspore/lite/test/models_onnx_fp16.cfg +++ b/mindspore/lite/test/models_onnx_fp16.cfg @@ -25,7 +25,7 @@ mnist-8.onnx 10 #super-resolution-10.onnx 1 #tinyyolov2-8.onnx 0.3 ml_2012_ocr_cn.onnx 200 -#ml_2012_ocr_cn_noLSTM.onnx 0.1 +ml_2012_ocr_cn_noLSTM.onnx 1 candy-9.onnx 5 mosaic-9.onnx 4 pointilism-9.onnx 3