From e428c6cf1f3487534e5595094da268ef9481b2ee Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=EA=B9=80=EC=A7=84=ED=98=84?= Date: Fri, 4 Oct 2024 15:42:37 +0900 Subject: [PATCH] =?UTF-8?q?Feat:=20=EB=AA=A8=EB=8D=B8=20=ED=95=99=EC=8A=B5?= =?UTF-8?q?=20patience(=EC=96=BC=EB=A6=AC=EC=8A=A4=ED=86=B1)=20=EC=A0=9C?= =?UTF-8?q?=EA=B1=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ai/app/api/yolo/classfication.py | 3 ++- ai/app/api/yolo/detection.py | 3 ++- ai/app/api/yolo/segmentation.py | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) 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 27a80cb..5785520 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 71203f1..0e215ec 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: # 콜백 해제 및 자원 해제