From 91ed16ba7ec96ccfe0dee27ca8923d566d28371a Mon Sep 17 00:00:00 2001 From: Jeff Donahue Date: Sat, 26 Jul 2014 20:57:07 -0700 Subject: [PATCH] Move loss display before lr display in ComputeUpdateValue. --- src/caffe/solver.cpp | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/caffe/solver.cpp b/src/caffe/solver.cpp index b44948e..9173881 100644 --- a/src/caffe/solver.cpp +++ b/src/caffe/solver.cpp @@ -102,18 +102,20 @@ void Solver::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(); -- 2.7.4