fix det distill yml
@@ -19,28 +19,26 @@ Architecture:
name: DistillationModel
algorithm: Distillation
Models:
- Student:
- freeze_params: false
+ Teacher:
+ freeze_params: true
return_all_feats: false
model_type: det
algorithm: DB
+ Transform:
Backbone:
- name: MobileNetV3
- scale: 0.5
- model_name: large
- disable_se: True
+ name: ResNet
+ layers: 18
Neck:
name: DBFPN
- out_channels: 96
+ out_channels: 256
Head:
name: DBHead
k: 50
- Student2:
+ Student:
freeze_params: false
return_all_feats: false
model_type: det
algorithm: DB
- Transform:
Backbone:
name: MobileNetV3
scale: 0.5
@@ -52,18 +50,20 @@ Architecture:
Head:
name: DBHead
k: 50
- Teacher:
- freeze_params: true
+ Student2:
+ freeze_params: false
return_all_feats: false
model_type: det
algorithm: DB
Transform:
Backbone:
- name: ResNet
- layers: 18
+ name: MobileNetV3
+ scale: 0.5
+ model_name: large
+ disable_se: True
Neck:
name: DBFPN
- out_channels: 256
+ out_channels: 96
Head:
name: DBHead
k: 50
