diff --git a/parser/caffe/caffe_parser.cc b/parser/caffe/caffe_parser.cc index 455bae9..af9de49 100644 --- a/parser/caffe/caffe_parser.cc +++ b/parser/caffe/caffe_parser.cc @@ -282,9 +282,6 @@ Status CaffeModelParser::ParseInput(domi::caffe::NetParameter &proto_message, bo return FAILED; } int input_dim_size = proto_message.input_dim_size(); - GE_CHK_BOOL_TRUE_EXEC_WITH_LOG((proto_message.input_size() == 0), - ErrorManager::GetInstance().ATCReportErrMessage("E11002"); - return PARAM_INVALID, "[Check][Size]Model has no input."); GE_CHK_BOOL_TRUE_EXEC_WITH_LOG((input_dim_size / proto_message.input_size() != parser::DIM_DEFAULT_SIZE || input_dim_size % proto_message.input_size() != 0), @@ -371,8 +368,11 @@ Status CaffeModelParser::ParseInput(domi::caffe::NetParameter &proto_message, bo input_data_flag = true; } } + } else { + ErrorManager::GetInstance().ATCReportErrMessage("E11002"); + GELOGE(FAILED, "[Check][Size]Model has no input."); + return PARAM_INVALID; } - return SUCCESS; }