diff --git a/src/training/trainer_new.py b/src/training/trainer_new.py
index 33188e9..30d706d 100644
--- a/src/training/trainer_new.py
+++ b/src/training/trainer_new.py
@@ -244,6 +244,7 @@ class Trainer:
         self._restore_from_resume()
         self._setup_profiler()
 
+        #! ------ passed: OK --------------
         try:
             self._train_loop()
             self._final_evaluation()