From 97cd241872c8bf0c94efdca29f279ac8a70bc801 Mon Sep 17 00:00:00 2001
From: "Jin, Ge" <ge.jin@intel.com>
Date: Tue, 4 Jul 2017 14:45:56 -0400
Subject: [PATCH] Fix bug in RemoveBNScale

Signed-off-by: Jin, Ge <ge.jin@intel.com>
---
 src/caffe/net.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp
index a7bbe8170..19e3dd7e1 100644
--- a/src/caffe/net.cpp
+++ b/src/caffe/net.cpp
@@ -1280,8 +1280,8 @@ void Net<Dtype>::ShareTrainedLayersWith(const Net* other) {
     }
     //temp_net_param.mutable_compile_net_state()->set_bn_top_rename(other->bn_top_rename_);
     complete_net_param.CopyFrom(temp_net_param);
-    complete_net_param.clear_layer();
     if (other->bn_scale_merge_) {
+      complete_net_param.clear_layer();
       RecoverBNScaleMergedNet<Dtype>(&temp_net_param, &complete_net_param);
     }
     CopyTrainedLayersFrom(complete_net_param);