Add post step call instead of solution_accepted

polyfem · Dec 6, 2023 · 148752f · 148752f
1 parent ab7091d
commit 148752f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/polysolve/nonlinear/Solver.cpp b/src/polysolve/nonlinear/Solver.cpp
@@ -181,6 +181,8 @@ namespace polysolve::nonlinear
             objFunc.solution_changed(x);
         }
 
+        objFunc.post_step(this->m_current.iterations, x);
+
         const auto g_norm_tol = this->m_stop.gradNorm;
         this->m_stop.gradNorm = first_grad_norm_tol;
 
@@ -195,7 +197,6 @@ namespace polysolve::nonlinear
             this->m_stop.fDelta, this->m_stop.gradNorm, this->m_stop.xDelta);
 
         update_solver_info(objFunc.value(x));
-        objFunc.step_accepted(this->m_current.iterations, x);
 
         int f_delta_step_cnt = 0;
         double f_delta = 0;
@@ -374,7 +375,6 @@ namespace polysolve::nonlinear
                 this->m_status = cppoptlib::Status::IterationLimit;
 
             update_solver_info(energy);
-            objFunc.step_accepted(this->m_current.iterations, x);
 
             // reset the tolerance, since in the first iter it might be smaller
             this->m_stop.gradNorm = g_norm_tol;