From f881d921895570987d101a898e750dd7d6e3080b Mon Sep 17 00:00:00 2001 From: wangxiaotian22 Date: Mon, 31 May 2021 15:00:19 +0800 Subject: [PATCH] modify --- parser/caffe/caffe_parser.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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; }