diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp index f1fc63aba95..89d14013dc9 100644 --- a/src/caffe/net.cpp +++ b/src/caffe/net.cpp @@ -182,7 +182,7 @@ void Net::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);