Update train.py

2025-03-25 08:51:51 +07:00 · 2025-03-25 08:51:51 +07:00 · 052e385473
parent f770973b5e
commit 052e385473
1 changed files with 2 additions and 0 deletions
--- a/src/server/train.py
+++ b/src/server/train.py
@ -61,6 +61,8 @@ elif os.path.exists(image_folder_yesterday) and os.path.exists(label_folder_yest
 else:
    # print(image_folder)
    # print(label_folder)
+    with open(LOG_FILE, "a") as log:
+        log.write(f"\nDate: {today_str}. No valid image & label folder found in the last two days!")
    raise Exception("  No valid image & label folder found in the last two days!")

 # 🏗️ Create dataset folders