diff --git a/src/server/train.py b/src/server/train.py index 87c4146..43689f5 100755 --- 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