diff --git a/ai/app/api/yolo/classfication.py b/ai/app/api/yolo/classfication.py index 592fd98..3de0eea 100644 --- a/ai/app/api/yolo/classfication.py +++ b/ai/app/api/yolo/classfication.py @@ -179,7 +179,8 @@ def run_train(request, model, dataset_root_path): batch=request.batch, lr0=request.lr0, lrf=request.lrf, - optimizer=request.optimizer + optimizer=request.optimizer, + patience=0 ) finally: # 콜백 해제 및 자원 해제 diff --git a/ai/app/api/yolo/detection.py b/ai/app/api/yolo/detection.py index dc7565c..0fc4c62 100644 --- a/ai/app/api/yolo/detection.py +++ b/ai/app/api/yolo/detection.py @@ -261,7 +261,8 @@ def run_train(request, model, dataset_root_path): batch=request.batch, lr0=request.lr0, lrf=request.lrf, - optimizer=request.optimizer + optimizer=request.optimizer, + patience=0 ) finally: # 콜백 해제 및 자원 해제 diff --git a/ai/app/api/yolo/segmentation.py b/ai/app/api/yolo/segmentation.py index 0763a58..c372cd0 100644 --- a/ai/app/api/yolo/segmentation.py +++ b/ai/app/api/yolo/segmentation.py @@ -205,7 +205,8 @@ def run_train(request, model, dataset_root_path): batch=request.batch, lr0=request.lr0, lrf=request.lrf, - optimizer=request.optimizer + optimizer=request.optimizer, + patience=0 ) finally: # 콜백 해제 및 자원 해제