diff --git a/train/src/main.py b/train/src/main.py
index 20e7038..5e1fc40 100644
--- a/train/src/main.py
+++ b/train/src/main.py
@@ -1332,6 +1332,7 @@ def stop_on_batch_end_if_needed(trainer_validator, *args, **kwargs):
             **additional_params,
         )
 
+    print(app.is_stopped())
     progress_bar_iters.hide()
     progress_bar_epochs.hide()
     watcher.running = False