/caffe

display 'ignore source layer' when initializing from existing parameters #3305

Merged
merged 1 commit into from Nov 10, 2015
Commits
Jump to file or symbol
Failed to load files and symbols.
+3 −3
Split
View
6 src/caffe/net.cpp
@@ -745,7 +745,7 @@ void Net<Dtype>::ShareTrainedLayersWith(const Net* other) {
++target_layer_id;
}
if (target_layer_id == layer_names_.size()) {
- DLOG(INFO) << "Ignoring source layer " << source_layer_name;
+ LOG(INFO) << "Ignoring source layer " << source_layer_name;
continue;
}
DLOG(INFO) << "Copying source layer " << source_layer_name;
@@ -813,7 +813,7 @@ void Net<Dtype>::CopyTrainedLayersFrom(const NetParameter& param) {
++target_layer_id;
}
if (target_layer_id == layer_names_.size()) {
- DLOG(INFO) << "Ignoring source layer " << source_layer_name;
+ LOG(INFO) << "Ignoring source layer " << source_layer_name;
continue;
}
DLOG(INFO) << "Copying source layer " << source_layer_name;
@@ -868,7 +868,7 @@ void Net<Dtype>::CopyTrainedLayersFromHDF5(const string trained_filename) {
for (int i = 0; i < num_layers; ++i) {
string source_layer_name = hdf5_get_name_by_idx(data_hid, i);
if (!layer_names_index_.count(source_layer_name)) {
- DLOG(INFO) << "Ignoring source layer " << source_layer_name;
+ LOG(INFO) << "Ignoring source layer " << source_layer_name;
continue;
}
int target_layer_id = layer_names_index_[source_layer_name];