!4943 process the bn with phase in prototxt

Merge pull request !4943 from zhaodezan/master
pull/4943/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 6a5c517d7b

@ -190,6 +190,14 @@ STATUS CaffeModelParser::ParseLayer(const caffe::NetParameter &proto,
continue;
}
// here we only process the bn with phase
if (layer.type() == "BatchNorm" && layer.include_size() == 1) {
if (layer.include(0).phase() == caffe::TRAIN) {
MS_LOG(INFO) << "Skip layer " << layer.name();
continue;
}
}
std::unique_ptr<schema::CNodeT> op = std::make_unique<schema::CNodeT>();
op->name = layer.name();

Loading…
Cancel
Save