From 3d6d2150e4d3a5ecf7b57efe00f10b036ea36c4b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Miroslav=20Kab=C3=A1t?= Date: Tue, 21 Sep 2021 01:54:41 +0200 Subject: [PATCH] Init _bestMetricValue for Loss metric _bestMetricValue must be initialized with Single.MaxValue for Loss metric --- src/Microsoft.ML.Vision/ImageClassificationTrainer.cs | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/Microsoft.ML.Vision/ImageClassificationTrainer.cs b/src/Microsoft.ML.Vision/ImageClassificationTrainer.cs index 7824838dea..3b8d6f62f5 100644 --- 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; + } } ///