From 9e70ce51901b5b07521a4508653697d0ac72440e Mon Sep 17 00:00:00 2001 From: Yangqing Jia Date: Wed, 23 Oct 2013 14:56:06 -0700 Subject: [PATCH] net update --- src/caffe/net.cpp | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp index 50f4f93..38a806d 100644 --- a/src/caffe/net.cpp +++ b/src/caffe/net.cpp @@ -96,8 +96,13 @@ Net::Net(const NetParameter& param, } } // After this layer is connected, set it up. - LOG(INFO) << "Setting up " << layer_names_[i]; + // LOG(INFO) << "Setting up " << layer_names_[i]; layers_[i]->SetUp(bottom_vecs_[i], &top_vecs_[i]); + for (int topid = 0; topid < top_vecs_[i].size(); ++topid) { + LOG(INFO) << "Top shape: " << top_vecs_[i][topid]->channels() << " " + << top_vecs_[i][topid]->height() << " " + << top_vecs_[i][topid]->width(); + } // Check if this layer needs backward operation itself for (int j = 0; j < layers_[i]->layer_param().blobs_lr_size(); ++j) { need_backward |= (layers_[i]->layer_param().blobs_lr(j) > 0); @@ -160,11 +165,6 @@ void Net::GetLearningRateAndWeightDecay() { params_weight_decay_.push_back(1.); } } - for (int topid = 0; topid < top_vecs_[i].size(); ++topid) { - LOG(INFO) << "Top shape: " << top_vecs_[i][topid]->channels() << " " - << top_vecs_[i][topid]->height() << " " - << top_vecs_[i][topid]->width(); - } } } -- 2.7.4