change runs dir

supervisely-ecosystem · Sep 14, 2023 · 621d31c · 621d31c
1 parent d1c7323
commit 621d31c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/supervisely/train/src/sly_train_globals.py b/supervisely/train/src/sly_train_globals.py
@@ -48,11 +48,10 @@
     finetune_str = file.read()  # yaml.safe_load(
 
 
-runs_dir = os.path.join(my_app.data_dir, 'runs')
+runs_dir = os.path.join(sly.app.get_data_dir(), 'runs')
 sly.fs.mkdir(runs_dir, remove_content_if_exists=True)  # for debug, does nothing in production
 experiment_name = str(task_id)
-# local_artifacts_dir = os.path.join(runs_dir, experiment_name)
-local_artifacts_dir = sly.app.get_data_dir()
+local_artifacts_dir = os.path.join(runs_dir, experiment_name)
 sly.logger.info(f"All training artifacts will be saved to local directory {local_artifacts_dir}")
 remote_artifacts_dir = os.path.join("/yolov5_train", project_info.name, experiment_name)
 remote_artifacts_dir = api.file.get_free_dir_name(team_id, remote_artifacts_dir)