Browse Source

!7117 [MSLITE] add mobilefacenet model

Merge pull request !7117 from ghzl/fix-mobileface-net
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
198dc2bce0
2 changed files with 2 additions and 5 deletions
  1. +1
    -0
      mindspore/lite/test/models_mindspore.cfg
  2. +1
    -5
      mindspore/lite/tools/converter/legacy_optimizer/graph/trans_format_insert_pass.cc

+ 1
- 0
mindspore/lite/test/models_mindspore.cfg View File

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

+ 1
- 5
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;
}



Loading…
Cancel
Save