add debug logs

train/src/callbacks.py

@@ -1,3 +1,4 @@
 def on_train_batch_end(trainer):
+    print("train batch end...")
     with open("train_batches.txt", "w") as file:
         file.write("train batch end")

train/src/main.py

@@ -1303,10 +1303,12 @@ def train_batch_watcher_disable():
         threading.Thread(target=train_batch_watcher_func, daemon=True).start()
 
     def stop_on_batch_end_if_needed(trainer_validator, *args, **kwargs):
+        print("Checking if app is stopped")
         try:
             app_is_stopped = app.is_stopped()
         except:
             app_is_stopped = True
+        print(f"App is stopped: {app_is_stopped}")
         if app_is_stopped:
             trainer_validator.stop = True
             raise app.StopException("This error is expected.")