Use net_->learnable_params() instead of net_->params() in RMSprop
authorRonghang Hu <huronghang@hotmail.com>
Sun, 9 Aug 2015 18:03:23 +0000 (11:03 -0700)
committerRonghang Hu <huronghang@hotmail.com>
Sun, 9 Aug 2015 18:03:23 +0000 (11:03 -0700)
commitb97ca6d4d17b4a661d553c233c2dfe03b88033c5
tree10285d07bbd3a8fad37eda8ebc681358aa88f56b
parent28a579eaf0668850705598b3075b8969f22226d9
Use net_->learnable_params() instead of net_->params() in RMSprop

In RMSProp solver, use const vector<Blob<Dtype>*>& net_params = this->net_->learnable_params(); instead of const vector<shared_ptr<Blob<Dtype> > >& net_params = this->net_->params();
src/caffe/solver.cpp