diff --git a/external/anomaly/ote_anomalib/callbacks/progress.py b/external/anomaly/ote_anomalib/callbacks/progress.py
index b6e1e8d316f..395da247ced 100644
--- a/external/anomaly/ote_anomalib/callbacks/progress.py
+++ b/external/anomaly/ote_anomalib/callbacks/progress.py
@@ -113,13 +113,13 @@ def _get_progress(self, stage: str = "train") -> float:
 
         return self._progress
     
-    def on_validation_epoch_end(self, trainer, pl_module):
-        super().on_validation_epoch_end(trainer, pl_module)
+    def on_train_epoch_end(self, trainer, pl_module):
+        super().on_train_epoch_end(trainer, pl_module)
         score = None
         metric = getattr(self.update_progress_callback, 'metric', None)
         if metric in trainer.logged_metrics:
             score = float(trainer.logged_metrics[metric])
-        progress = int(self._get_progress('predict'))
+        progress = int(self._get_progress('train'))
         self.update_progress_callback(progress=progress, score=score)
 
     def _update_progress(self, stage: str):