This commit is contained in:
Double_V
2022-05-01 19:54:22 +08:00
committed by GitHub
parent 77703f37fb
commit 6c68808a3f
@@ -37,7 +37,7 @@ Architecture:
Head:
name: DBHead
k: 50
Student2:
Teacher:
pretrained: ./pretrain_models/MobileNetV3_large_x0_5_pretrained
freeze_params: false
return_all_feats: false
@@ -62,15 +62,15 @@ Loss:
loss_config_list:
- DistillationDMLLoss:
model_name_pairs:
- ["Student", "Student2"]
- ["Student", "Teacher"]
maps_name: "thrink_maps"
weight: 1.0
act: "softmax"
model_name_pairs: ["Student", "Student2"]
model_name_pairs: ["Student", "Teacher"]
key: maps
- DistillationDBLoss:
weight: 1.0
model_name_list: ["Student", "Student2"]
model_name_list: ["Student", "Teacher"]
# key: maps
name: DBLoss
balance_loss: true
@@ -94,7 +94,7 @@ Optimizer:
PostProcess:
name: DistillationDBPostProcess
model_name: ["Student", "Student2"]
model_name: ["Student", "Teacher"]
key: head_out
thresh: 0.3
box_thresh: 0.6