From 5edf3419f97d252fc30480e0c5180e2cedbe2036 Mon Sep 17 00:00:00 2001 From: Victor Costa Date: Thu, 13 Feb 2025 04:40:57 +0000 Subject: [PATCH] Fix updating of best epoch during early stopping (#19164) --- ultralytics/utils/torch_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/utils/torch_utils.py b/ultralytics/utils/torch_utils.py index 305dda50..00aa8df5 100644 --- a/ultralytics/utils/torch_utils.py +++ b/ultralytics/utils/torch_utils.py @@ -746,7 +746,7 @@ class EarlyStopping: if fitness is None: # check if fitness=None (happens when val=False) return False - if fitness >= self.best_fitness: # >= 0 to allow for early zero-fitness stage of training + if fitness > self.best_fitness or self.best_fitness == 0: # allow for early zero-fitness stage of training self.best_epoch = epoch self.best_fitness = fitness delta = epoch - self.best_epoch # epochs without improvement