diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp index 1b738c3d8..a7bbe8170 100644 --- a/src/caffe/net.cpp +++ b/src/caffe/net.cpp @@ -839,7 +839,9 @@ void Net::RemoveBNScale(const NetParameter& param, shared_ptr scale_layer_param(new LayerParameter()); RecoverScaleFromBN(child_layer_param, *scale_layer_param, (Dtype)1, (Dtype)0); AdjustConvLayer(*layer_param, child_layer_param, *scale_layer_param, is_net_init); - } + } else { + AdjustConvLayer(*layer_param, child_layer_param, grandchild_layer_param, true); + } if (bn_scale_remove == false) bn_scale_remove = true; layers_to_drop.insert(child_layer_param.name()); }