未验证 提交 6c68808a 编写于 作者: D Double_V 提交者: GitHub

fix dml (#6067)

上级 77703f37
...@@ -37,7 +37,7 @@ Architecture: ...@@ -37,7 +37,7 @@ Architecture:
Head: Head:
name: DBHead name: DBHead
k: 50 k: 50
Student2: Teacher:
pretrained: ./pretrain_models/MobileNetV3_large_x0_5_pretrained pretrained: ./pretrain_models/MobileNetV3_large_x0_5_pretrained
freeze_params: false freeze_params: false
return_all_feats: false return_all_feats: false
...@@ -62,15 +62,15 @@ Loss: ...@@ -62,15 +62,15 @@ Loss:
loss_config_list: loss_config_list:
- DistillationDMLLoss: - DistillationDMLLoss:
model_name_pairs: model_name_pairs:
- ["Student", "Student2"] - ["Student", "Teacher"]
maps_name: "thrink_maps" maps_name: "thrink_maps"
weight: 1.0 weight: 1.0
act: "softmax" act: "softmax"
model_name_pairs: ["Student", "Student2"] model_name_pairs: ["Student", "Teacher"]
key: maps key: maps
- DistillationDBLoss: - DistillationDBLoss:
weight: 1.0 weight: 1.0
model_name_list: ["Student", "Student2"] model_name_list: ["Student", "Teacher"]
# key: maps # key: maps
name: DBLoss name: DBLoss
balance_loss: true balance_loss: true
...@@ -94,7 +94,7 @@ Optimizer: ...@@ -94,7 +94,7 @@ Optimizer:
PostProcess: PostProcess:
name: DistillationDBPostProcess name: DistillationDBPostProcess
model_name: ["Student", "Student2"] model_name: ["Student", "Teacher"]
key: head_out key: head_out
thresh: 0.3 thresh: 0.3
box_thresh: 0.6 box_thresh: 0.6
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册