Made the stopping condition for the structural svm solver a little more robust.

4d5249df · Davis King · 20dde81b · 4d5249df
Commit 4d5249df authored Jan 05, 2014 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

structural_svm_problem.h dlib/svm/structural_svm_problem.h +4 -2

No files found.
--- a/dlib/svm/structural_svm_problem.h
+++ b/dlib/svm/structural_svm_problem.h
@@ -438,7 +438,8 @@ namespace dlib

            if (converged)
            {
-                return current_risk_gap < std::max(cache_based_eps,cache_based_eps*current_risk_value);
+                return (current_risk_gap < std::max(cache_based_eps,cache_based_eps*current_risk_value)) || 
+                       (current_risk_gap == 0);
            }

            if (current_risk_gap < eps)
@@ -452,7 +453,8 @@ namespace dlib
                {
                    converged = true;
                    skip_cache = false;
-                    return current_risk_gap < std::max(cache_based_eps,cache_based_eps*current_risk_value);
+                    return (current_risk_gap < std::max(cache_based_eps,cache_based_eps*current_risk_value));
+                           (current_risk_gap == 0);
                }

                ++count_below_eps;