glenn-jocher commited on
Commit
fd71fe8
1 Parent(s): f7e075f

Update hyp.finetune.yaml

Browse files
Files changed (1) hide show
  1. 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.2 # cls loss gain
11
  cls_pw: 1.0 # cls BCELoss positive_weight
12
- obj: 0.3 # 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
 
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