From 99043530b350e6b9499029b4dfd8d1ae95f7bf26 Mon Sep 17 00:00:00 2001
From: almaz <almazgimaev@gmail.com>
Date: Fri, 13 Dec 2024 14:59:08 +0100
Subject: [PATCH] fix path

---
 train/src/ui/monitoring.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train/src/ui/monitoring.py b/train/src/ui/monitoring.py
index 6893523..ce48915 100644
--- 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))