|
|
|
@ -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();
|
|
|
|
|
|
|
|
|
|