Do snapshot after computing loss and test accuracy

diff --git a/src/caffe/solver.cpp b/src/caffe/solver.cpp
index e02b72f..8195840 100644 (file)
--- a/src/caffe/solver.cpp
+++ b/src/caffe/solver.cpp
@@ -58,10 +58,6 @@ void Solver<Dtype>::Solve(const char* resume_file) {
     ComputeUpdateValue();
     net_->Update();
 
-    // Check if we need to do snapshot
-    if (param_.snapshot() && iter_ % param_.snapshot() == 0) {
-      Snapshot();
-    }
     if (param_.display() && iter_ % param_.display() == 0) {
       LOG(INFO) << "Iteration " << iter_ << ", loss = " << loss;
     }
@@ -71,6 +67,10 @@ void Solver<Dtype>::Solve(const char* resume_file) {
       Test();
       Caffe::set_phase(Caffe::TRAIN);
     }
+    // Check if we need to do snapshot
+    if (param_.snapshot() && iter_ % param_.snapshot() == 0) {
+      Snapshot();
+    }
   }
   // After the optimization is done, always do a snapshot.
   iter_--;