From cb22a5c612a40d05fbe67b0e62dc3efa41eabbb7 Mon Sep 17 00:00:00 2001 From: zengxianglong Date: Mon, 23 Nov 2020 10:26:39 +0800 Subject: [PATCH] add to onnx model --- mindspore/lite/src/lite_session.cc | 2 +- mindspore/lite/test/models_onnx.cfg | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/lite/src/lite_session.cc b/mindspore/lite/src/lite_session.cc index 7f5e43d2c61..d3d4b1f1102 100644 --- a/mindspore/lite/src/lite_session.cc +++ b/mindspore/lite/src/lite_session.cc @@ -199,7 +199,7 @@ void LiteSession::InitGraphInputMap(const lite::Model *model) { MS_ASSERT(in_node != nullptr); auto in_size = in_node->input_indices_.size(); for (size_t i = 0; i < in_size; ++i) { - MS_ASSERT(this->input_map_.find(in_node->name_) + std::to_string(i) == this->input_map_.end()); + MS_ASSERT(this->input_map_.find(in_node->name_ + std::to_string(i)) == this->input_map_.end()); auto in_tensor_index = size_t(in_node->input_indices_[i]); bool is_graph_input = false; for (size_t j = 0; j < graph_in_size; ++j) { diff --git a/mindspore/lite/test/models_onnx.cfg b/mindspore/lite/test/models_onnx.cfg index 2ce2c6d8252..ce7db58afa9 100644 --- a/mindspore/lite/test/models_onnx.cfg +++ b/mindspore/lite/test/models_onnx.cfg @@ -32,3 +32,5 @@ pointilism-9.onnx rain-princess-9.onnx udnie-9.onnx adversarial_pruning.onnx +residual_distill_res34_cifar10_bs_1_update.onnx +residual_distill_res50_cifar10_bs_1_update.onnx