glenn-jocher
commited on
Commit
•
fd71fe8
1
Parent(s):
f7e075f
Update hyp.finetune.yaml
Browse files- data/hyp.finetune.yaml +2 -2
data/hyp.finetune.yaml
CHANGED
@@ -7,9 +7,9 @@ lr0: 0.01 # initial learning rate (SGD=1E-2, Adam=1E-3)
|
|
7 |
momentum: 0.94 # SGD momentum/Adam beta1
|
8 |
weight_decay: 0.0005 # optimizer weight decay 5e-4
|
9 |
giou: 0.05 # GIoU loss gain
|
10 |
-
cls: 0.
|
11 |
cls_pw: 1.0 # cls BCELoss positive_weight
|
12 |
-
obj: 0.
|
13 |
obj_pw: 1.0 # obj BCELoss positive_weight
|
14 |
iou_t: 0.20 # IoU training threshold
|
15 |
anchor_t: 4.0 # anchor-multiple threshold
|
|
|
7 |
momentum: 0.94 # SGD momentum/Adam beta1
|
8 |
weight_decay: 0.0005 # optimizer weight decay 5e-4
|
9 |
giou: 0.05 # GIoU loss gain
|
10 |
+
cls: 0.4 # cls loss gain
|
11 |
cls_pw: 1.0 # cls BCELoss positive_weight
|
12 |
+
obj: 0.5 # obj loss gain (scale with pixels)
|
13 |
obj_pw: 1.0 # obj BCELoss positive_weight
|
14 |
iou_t: 0.20 # IoU training threshold
|
15 |
anchor_t: 4.0 # anchor-multiple threshold
|