yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add to onnx model

This commit is contained in:
zengxianglong 2020-11-23 10:26:39 +08:00
parent 7aac8c0b89
commit cb22a5c612
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/src/lite_session.cc
View File

@ -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) {

2
mindspore/lite/test/models_onnx.cfg
View File

@ -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