Browse Source

bugfix for onnx parser optional input

pull/201/head
y00500818 5 years ago
parent
commit
697941c424
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      parser/onnx/onnx_parser.cc

+ 4
- 2
parser/onnx/onnx_parser.cc View File

@@ -334,9 +334,11 @@ Status OnnxModelParser::SetOperatorInputs() {
for (auto in_iter = inputs_map_.begin(); in_iter != inputs_map_.end(); in_iter++) { for (auto in_iter = inputs_map_.begin(); in_iter != inputs_map_.end(); in_iter++) {
auto out_iter = outputs_map_.find(in_iter->first); auto out_iter = outputs_map_.find(in_iter->first);
if (out_iter == outputs_map_.end()) { if (out_iter == outputs_map_.end()) {
GELOGE(INTERNAL_ERROR, "Unknown input: %s:%d in node: %s", in_iter->first.c_str(), in_iter->second[0].second,
GELOGW("Unknown input: %s:%d for node: %s, which maybe option input.",
in_iter->first.c_str(),
in_iter->second[0].second,
in_iter->second[0].first.c_str()); in_iter->second[0].first.c_str());
return INTERNAL_ERROR;
continue;
} }


std::vector<std::pair<std::string, int>> &input_node_indexs = in_iter->second; std::vector<std::pair<std::string, int>> &input_node_indexs = in_iter->second;


Loading…
Cancel
Save