From a71354f87d9b4cf205180e5339b958fc684e8a51 Mon Sep 17 00:00:00 2001 From: Evan Shelhamer Date: Thu, 26 Jun 2014 13:05:44 -0700 Subject: [PATCH] rename layer -> param mapping for clarity --- include/caffe/net.hpp | 2 +- src/caffe/net.cpp | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/include/caffe/net.hpp b/include/caffe/net.hpp index aa540ed..cbd5bec 100644 --- a/include/caffe/net.hpp +++ b/include/caffe/net.hpp @@ -144,7 +144,7 @@ class Net { vector*> > top_vecs_; vector > top_id_vecs_; vector param_owners_; - vector > param_net_indices_; + vector > layer_param_indices_; map param_names_index_; // blob indices for the input and the output of the net vector net_input_blob_indices_; diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp index e25c754..f364e67 100644 --- a/src/caffe/net.cpp +++ b/src/caffe/net.cpp @@ -242,7 +242,7 @@ void Net::AppendParam(const NetParameter& param, const int layer_id, } const int net_param_id = params_.size(); params_.push_back(layers_[layer_id]->blobs()[param_id]); - param_net_indices_.push_back(make_pair(layer_id, param_id)); + layer_param_indices_.push_back(make_pair(layer_id, param_id)); if (!param_size || !param_name.size() || (param_name.size() && param_names_index_.find(param_name) == param_names_index_.end())) { // This layer "owns" this parameter blob -- it is either anonymous @@ -257,7 +257,7 @@ void Net::AppendParam(const NetParameter& param, const int layer_id, const int owner_net_param_id = param_names_index_[param_name]; param_owners_.push_back(owner_net_param_id); const pair& owner_index = - param_net_indices_[owner_net_param_id]; + layer_param_indices_[owner_net_param_id]; const int owner_layer_id = owner_index.first; const int owner_param_id = owner_index.second; LOG(INFO) << "Sharing parameters '" << param_name << "' owned by " -- 2.7.4