fix path

supervisely-ecosystem · Dec 13, 2024 · 9904353 · 9904353
1 parent 1435e4a
commit 9904353
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train/src/ui/monitoring.py b/train/src/ui/monitoring.py
@@ -667,7 +667,7 @@ def train(api: sly.Api, task_id, context, state, app_logger):
         g.api.app.set_fields(g.task_id, fields)
 
         if state["saveLast"] is False:
-            for root, _, files in os.walk(cfg.work_dir):
+            for root, _, files in os.walk(g.checkpoints_dir):
                 for file in files:
                     if file == "latest.pth":
                         sly.fs.silent_remove(os.path.join(root, file))