From fef22a3cf6356048a2f7f6dc0ec6975c7c980314 Mon Sep 17 00:00:00 2001 From: nguentrungthat Date: Fri, 14 Mar 2025 16:58:46 +0700 Subject: [PATCH] Update train.py --- src/server/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/server/train.py b/src/server/train.py index 21b0a7b..3994630 100755 --- a/src/server/train.py +++ b/src/server/train.py @@ -104,7 +104,7 @@ def get_latest_model(trained_model_folder: str, default_model: str = "train5/wei # Lấy danh sách các thư mục theo ngày, loại bỏ thư mục của ngày hôm nay subfolders = [ f for f in os.listdir(trained_model_folder) - if os.path.isdir(os.path.join(trained_model_folder, f)) and f < today_str + if os.path.isdir(os.path.join(trained_model_folder, f)) and (f == today_str or f.startswith(today_str)) ] subfolders = sorted(subfolders, reverse=True) # Sắp xếp giảm dần (mới nhất trước) @@ -248,6 +248,6 @@ else: train_yolo_model(pretrained_model = get_latest_model( trained_model_folder=TRAINED_MODEL_FOLDER, default_model=PRETRAINED_MODEL), - dataset_folder = dataset_folder, epochs = 2, name=today_str, project_name=model_folder_name + dataset_folder = dataset_folder, epochs = 50, name=today_str, project_name=model_folder_name )