Fix bug in RemoveBNScale

To handle "conv----bn" structure Signed-off-by: Jin, Ge <[email protected]>
intel · Jul 3, 2017 · 881163a · 881163a
1 parent 792b0d1
commit 881163a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp
@@ -839,7 +839,9 @@ void Net<Dtype>::RemoveBNScale(const NetParameter& param,
             shared_ptr<LayerParameter> scale_layer_param(new LayerParameter());
             RecoverScaleFromBN(child_layer_param, *scale_layer_param, (Dtype)1, (Dtype)0);
             AdjustConvLayer<Dtype>(*layer_param, child_layer_param, *scale_layer_param, is_net_init);
-          }
+          } else {
+            AdjustConvLayer<Dtype>(*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());
         }