🎨 [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:
|