From: shai <shai@magisto.com>
Date: Sun, 15 Jan 2017 08:54:45 +0000 (+0000)
Subject: fixing upgrade_proto for BatchNorm layer: be more conservative leave "name" in param... 
X-Git-Tag: submit/tizen/20180823.020014~95^2
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=a19357a190664b1ea99d18e14eedc27e43ebed42;p=platform%2Fupstream%2Fcaffeonacl.git

fixing upgrade_proto for BatchNorm layer: be more conservative leave "name" in param, only set lr_mult and decay_mult to zero
---

diff --git a/src/caffe/util/upgrade_proto.cpp b/src/caffe/util/upgrade_proto.cpp
index a0aacbe..94771c8 100644
--- a/src/caffe/util/upgrade_proto.cpp
+++ b/src/caffe/util/upgrade_proto.cpp
@@ -1018,7 +1018,13 @@ void UpgradeNetBatchNorm(NetParameter* net_param) {
     // the previous BatchNorm layer definition.
     if (net_param->layer(i).type() == "BatchNorm"
         && net_param->layer(i).param_size() == 3) {
-      net_param->mutable_layer(i)->clear_param();
+      // set lr_mult and decay_mult to zero. leave all other param intact.
+      for (int ip = 0; ip < net_param->layer(i).param_size(); ip++) {
+        ParamSpec* fixed_param_spec =
+          net_param->mutable_layer(i)->mutable_param(ip);
+        fixed_param_spec->set_lr_mult(0.f);
+        fixed_param_spec->set_decay_mult(0.f);
+      }
     }
   }
 }