Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix mobilefacenet bug

This commit is contained in:
guohongzilong 2020-10-09 15:38:39 +08:00
parent b7183ded66
commit b42f38025c
2 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/lite/test/models_mindspore.cfg
View File

@ -5,4 +5,5 @@ shufflenetv2.mindir
inceptionv3.mindir
googlenet.mindir
retinaface.mindir
mobilefacenet.mindir
resnext50.mindir

6
mindspore/lite/tools/converter/legacy_optimizer/graph/trans_format_insert_pass.cc
View File

@ -93,11 +93,7 @@ bool TransOpInsertPass::CanFusion(schema::MetaGraphT *graph, const std::unique_p
return has_trans_count >= half_count;
}
}
if (total_node_count % 2 == 0) {
can_fusion = has_trans_count > half_count;
} else {
can_fusion = has_trans_count >= half_count;
}
can_fusion = has_trans_count > half_count;
return can_fusion;
}