Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "b25843e799d42246d7a60808259dc5c28163446f"
Commit fd628d56 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

fix det distill yml

parent fffd556c
Show whitespace changes
Inline Side-by-side
Showing with 15 additions and 15 deletions
configs/det/ch_ppocr_v2.1/ch_det_lite_train_cml_v2.1.yml
View file @ fd628d56
...@@ -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,18 +50,20 @@ Architecture: ...@@ -52,18 +50,20 @@ 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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment