Move loss display before lr display in ComputeUpdateValue.
authorJeff Donahue <jeff.donahue@gmail.com>
Sun, 27 Jul 2014 03:57:07 +0000 (20:57 -0700)
committerJeff Donahue <jeff.donahue@gmail.com>
Sun, 27 Jul 2014 19:05:06 +0000 (12:05 -0700)
src/caffe/solver.cpp

index b44948e..9173881 100644 (file)
@@ -102,18 +102,20 @@ void Solver<Dtype>::Solve(const char* resume_file) {
         iter_ % param_.snapshot() == 0) {
       Snapshot();
     }
+
     if (param_.test_interval() && iter_ % param_.test_interval() == 0) {
       TestAll();
     }
+
     const bool display = param_.display() && iter_ % param_.display() == 0;
     net_->set_debug_info(display && param_.debug_info());
     Dtype loss = net_->ForwardBackward(bottom_vec);
-    ComputeUpdateValue();
-    net_->Update();
-
     if (display) {
       LOG(INFO) << "Iteration " << iter_ << ", loss = " << loss;
     }
+
+    ComputeUpdateValue();
+    net_->Update();
   }
   // Always save a snapshot after optimization.
   Snapshot();