提交 fd628d56 编写于 作者: L LDOUBLEV

fix det distill yml

上级 fffd556c
...@@ -19,28 +19,26 @@ Architecture: ...@@ -19,28 +19,26 @@ Architecture:
name: DistillationModel name: DistillationModel
algorithm: Distillation algorithm: Distillation
Models: Models:
Student: Teacher:
freeze_params: false freeze_params: true
return_all_feats: false return_all_feats: false
model_type: det model_type: det
algorithm: DB algorithm: DB
Transform:
Backbone: Backbone:
name: MobileNetV3 name: ResNet
scale: 0.5 layers: 18
model_name: large
disable_se: True
Neck: Neck:
name: DBFPN name: DBFPN
out_channels: 96 out_channels: 256
Head: Head:
name: DBHead name: DBHead
k: 50 k: 50
Student2: Student:
freeze_params: false freeze_params: false
return_all_feats: false return_all_feats: false
model_type: det model_type: det
algorithm: DB algorithm: DB
Transform:
Backbone: Backbone:
name: MobileNetV3 name: MobileNetV3
scale: 0.5 scale: 0.5
...@@ -52,22 +50,24 @@ Architecture: ...@@ -52,22 +50,24 @@ Architecture:
Head: Head:
name: DBHead name: DBHead
k: 50 k: 50
Teacher: Student2:
freeze_params: true freeze_params: false
return_all_feats: false return_all_feats: false
model_type: det model_type: det
algorithm: DB algorithm: DB
Transform: Transform:
Backbone: Backbone:
name: ResNet name: MobileNetV3
layers: 18 scale: 0.5
model_name: large
disable_se: True
Neck: Neck:
name: DBFPN name: DBFPN
out_channels: 256 out_channels: 96
Head: Head:
name: DBHead name: DBHead
k: 50 k: 50
Loss: Loss:
name: CombinedLoss name: CombinedLoss
loss_config_list: loss_config_list:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册