Warning of fallback only the first time for cudnn_pooling_layer

BVLC · Dec 22, 2014 · 5c43f7e · 5c43f7e
1 parent 3e167c1
commit 5c43f7e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/caffe/layers/cudnn_pooling_layer.cu b/src/caffe/layers/cudnn_pooling_layer.cu
@@ -14,7 +14,7 @@ void CuDNNPoolingLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom,
     vector<Blob<Dtype>*>* top) {
   // Fallback to Caffe for padded pooling, max top mask.
   if ((this->pad_h_ > 0 || this->pad_w_ > 0) || (*top).size() > 1) {
-    LOG(WARNING) << "Falling back to standard Caffe for padded pooling.";
+    LOG_FIRST_N(WARNING,1) << "Falling back to standard Caffe for padded pooling.";
     return PoolingLayer<Dtype>::Forward_gpu(bottom, top);
   }
 
@@ -33,7 +33,7 @@ void CuDNNPoolingLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top,
 
   // Fallback to Caffe for padded pooling, max top mask.
   if ((this->pad_h_ > 0 || this->pad_w_ > 0) || top.size() > 1) {
-    LOG(WARNING) << "Falling back to standard Caffe for padded pooling.";
+    LOG_FIRST_N(WARNING,1) << "Falling back to standard Caffe for padded pooling.";
     return PoolingLayer<Dtype>::Backward_gpu(top, propagate_down, bottom);
   }
 

diff --git a/src/caffe/solver.cpp b/src/caffe/solver.cpp
@@ -176,7 +176,7 @@ void Solver<Dtype>::Solve(const char* resume_file) {
   CHECK_GE(average_loss, 1) << "average_cost should be non-negative.";
 
   vector<Dtype> losses;
-  Dtype smoothed_loss;
+  Dtype smoothed_loss = 0;
 
   // For a network that is trained by the solver, no bottom or top vecs
   // should be given, and we will just provide dummy vecs.