Init _bestMetricValue for Loss metric (#5939)

dotnet · Oct 5, 2021 · 9cdd778 · 9cdd778
1 parent 51c4984
commit 9cdd778
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/Microsoft.ML.Vision/ImageClassificationTrainer.cs b/src/Microsoft.ML.Vision/ImageClassificationTrainer.cs
@@ -258,7 +258,10 @@ public EarlyStopping(float minDelta = 0.01f, int patience = 20, EarlyStoppingMet
                 if (metric == EarlyStoppingMetric.Accuracy)
                     CheckIncreasing = true;
                 else if (metric == EarlyStoppingMetric.Loss)
+                {
                     CheckIncreasing = false;
+                    _bestMetricValue = Single.MaxValue;
+                }
             }
 
             /// <summary>