zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update hyp.finetune.yaml

This commit is contained in:
Glenn Jocher 2020-08-22 16:18:53 -07:00
parent f7e075fe93
commit fd71fe8451
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
data/hyp.finetune.yaml
View File

@ -7,9 +7,9 @@ lr0: 0.01 # initial learning rate (SGD=1E-2, Adam=1E-3)
momentum: 0.94 # SGD momentum/Adam beta1 momentum: 0.94 # SGD momentum/Adam beta1
weight_decay: 0.0005 # optimizer weight decay 5e-4 weight_decay: 0.0005 # optimizer weight decay 5e-4
giou: 0.05 # GIoU loss gain giou: 0.05 # GIoU loss gain
cls: 0.2 # cls loss gain cls: 0.4 # cls loss gain
cls_pw: 1.0 # cls BCELoss positive_weight cls_pw: 1.0 # cls BCELoss positive_weight
obj: 0.3 # obj loss gain (scale with pixels) obj: 0.5 # obj loss gain (scale with pixels)
obj_pw: 1.0 # obj BCELoss positive_weight obj_pw: 1.0 # obj BCELoss positive_weight
iou_t: 0.20 # IoU training threshold iou_t: 0.20 # IoU training threshold
anchor_t: 4.0 # anchor-multiple threshold anchor_t: 4.0 # anchor-multiple threshold