Merge pull request #3007 from philkr/neg_lr_mult

Compute backward for negative lr_mult
BVLC · Sep 1, 2015 · 2d55e31 · 2d55e31
2 parents d362894 + a675172
commit 2d55e31
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp
@@ -182,7 +182,7 @@ void Net<Dtype>::Init(const NetParameter& in_param) {
     for (int param_id = 0; param_id < num_param_blobs; ++param_id) {
       const ParamSpec* param_spec = (param_id < param_size) ?
           &layer_param.param(param_id) : &default_param_spec;
-      const bool param_need_backward = param_spec->lr_mult() > 0;
+      const bool param_need_backward = param_spec->lr_mult() != 0;
       need_backward |= param_need_backward;
       layers_[layer_id]->set_param_propagate_down(param_id,
                                                   param_need_backward);