diff --git a/supervisely/train/src/sly_train.py b/supervisely/train/src/sly_train.py
index 76946aa8afcc..4383af41d09f 100644
--- a/supervisely/train/src/sly_train.py
+++ b/supervisely/train/src/sly_train.py
@@ -128,7 +128,9 @@ def train(api: sly.Api, task_id, context, state, app_logger):
         upload_artifacts(g.local_artifacts_dir, g.remote_artifacts_dir)
         set_task_output()
     except Exception as e:
-        my_app.show_modal_window(f"Something went wrong. Find more info in the app logs. {repr(e)}", level="error")
+        msg = f"Something went wrong. Find more info in the app logs. {repr(e)}"
+        my_app.show_modal_window(msg, level="error")
+        sly.logger.error(msg, exc_info=True, extra={ 'exc_str': str(e)})
         api.app.set_field(task_id, "state.started", False)
 
     # stop application
@@ -150,7 +152,13 @@ def main():
     my_app.compile_template(g.root_source_dir)
 
     # init data for UI widgets
-    ui.init(data, state)
+    try:
+        ui.init(data, state)
+    except Exception as e:
+        msg = f"Data initialization error. Check your project and try again. {repr(e)}"
+        my_app.show_modal_window(msg, level="error")
+        sly.logger.error(msg, exc_info=True, extra={ 'exc_str': str(e)})
+        g.api.app.set_field(task_id, "state.started", False)
 
     my_app.run(data=data, state=state)