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

Merge branch 'ai/refactor/resource-cleanup' into 'ai/develop'

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

See merge request s11-s-project/S11P21S002!291
This commit is contained in:
김용수 2024-10-04 15:55:57 +09:00
commit dce2309f5b
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:
# 콜백 해제 및 자원 해제