🎨 [Update] Loss config
Browse files- 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 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
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:
|