henry000 commited on
Commit
73207bd
·
1 Parent(s): 77743d7

🎨 [Update] Loss config

Browse files
Files changed (1) hide show
  1. yolo/config/hyper/default.yaml +12 -13
yolo/config/hyper/default.yaml CHANGED
@@ -12,19 +12,18 @@ train:
12
  lr: 0.001
13
  weight_decay: 0.0001
14
  loss:
15
- BCELoss:
16
- args:
17
- BoxLoss:
18
- args:
19
- alpha: 0.1
20
- DFLoss:
21
- args:
22
- matcher:
23
- iou: CIoU
24
- topk: 10
25
- factor:
26
- iou: 6.0
27
- cls: 0.5
28
  scheduler:
29
  type: StepLR
30
  args:
 
12
  lr: 0.001
13
  weight_decay: 0.0001
14
  loss:
15
+ objective:
16
+ - [BCELoss, 0.1]
17
+ - [BoxLoss, 0.1]
18
+ - [DFLoss, 0.1]
19
+ aux:
20
+ True
21
+ matcher:
22
+ iou: CIoU
23
+ topk: 10
24
+ factor:
25
+ iou: 6.0
26
+ cls: 0.5
 
27
  scheduler:
28
  type: StepLR
29
  args: