jhynsoo/worlabel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Feat: 모델 학습 patience(얼리스톱) 제거

Browse Source
This commit is contained in:
김진현 2024-10-04 15:42:37 +09:00
parent fbc2c1b46a
commit e428c6cf1f
3 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ai/app/api/yolo/classfication.py
View File

@ -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:
# 콜백 해제 및 자원 해제

3
ai/app/api/yolo/detection.py
View File

@ -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:
# 콜백 해제 및 자원 해제

3
ai/app/api/yolo/segmentation.py
View File

@ -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:
# 콜백 해제 및 자원 해제