debug_info in NetParameter so it can be enabled outside training

diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp
index e827a06..4350e05 100644 (file)
--- a/src/caffe/net.cpp
+++ b/src/caffe/net.cpp
@@ -218,10 +218,9 @@ void Net<Dtype>::Init(const NetParameter& in_param) {
     layer_names_index_[layer_names_[layer_id]] = layer_id;
   }
   GetLearningRateAndWeightDecay();
+  debug_info_ = param.debug_info();
   LOG(INFO) << "Network initialization done.";
   LOG(INFO) << "Memory required for data: " << memory_used_ * sizeof(Dtype);
-  // Don't display debug info by default.
-  debug_info_ = false;
 }
 
 template <typename Dtype>

diff --git a/src/caffe/proto/caffe.proto b/src/caffe/proto/caffe.proto
index d6eec02..d65ef78 100644 (file)
--- a/src/caffe/proto/caffe.proto
+++ b/src/caffe/proto/caffe.proto
@@ -60,6 +60,10 @@ message NetParameter {
   // Some layers may be included/excluded depending on this state and the states
   // specified in the layers' include and exclude fields.
   optional NetState state = 6;
+
+  // Print debugging information about results while running Net::Forward,
+  // Net::Backward, and Net::Update.
+  optional bool debug_info = 7 [default = false];
 }
 
 // NOTE