Skip to content

GitLab

Commit b6cd6047 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

cml config

parent 293e85bc
Hide whitespace changes
Inline Side-by-side
Showing with 53 additions and 50 deletions
configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml
View file @ b6cd6047
...@@ -19,57 +19,60 @@ Global: ...@@ -19,57 +19,60 @@ Global:
distributed: true distributed: true
Architecture: Architecture:
name: DistillationModel
algorithm: Distillation
model_type: det model_type: det
algorithm: DB Models:
NetWorks: Student:
Student: model_type: det
model_type: det algorithm: DB
algorithm: DB Transform: null
Transform: null Backbone:
Backbone: name: MobileNetV3
name: MobileNetV3 scale: 0.5
scale: 0.5 model_name: large
model_name: large disable_se: true
disable_se: true Neck:
Neck: name: CAFPN
name: CAFPN out_channels: 96
out_channels: 96 shortcut: True
shortcut: True Head:
Head: name: DBHead
name: DBHead k: 50
k: 50 Student2:
Student2: model_type: det
model_type: det algorithm: DB
algorithm: DB Transform: null
Transform: null Backbone:
Backbone: name: MobileNetV3
name: MobileNetV3 scale: 0.5
scale: 0.5 model_name: large
model_name: large disable_se: true
disable_se: true Neck:
Neck: name: CAFPN
name: CAFPN out_channels: 96
out_channels: 96 shortcut: True
shortcut: True Head:
Head: name: DBHead
name: DBHead k: 50
k: 50 Teacher:
Teacher: pretrained: ./inference/ch_ppocr_resnet50/student
pretrained: ./inference/ch_ppocr_resnet50/student freeze_params: true
model_type: det return_all_feats: false
algorithm: DB model_type: det
Backbone: algorithm: DB
name: ResNet Backbone:
in_channels: 3 name: ResNet
layers: 50 in_channels: 3
Neck: layers: 50
name: FEPAN Neck:
out_channels: 256 name: FEPAN
Head: out_channels: 256
name: DBHead Head:
kernel_list: [7,2,2] name: DBHead
k: 50 kernel_list: [7,2,2]
k: 50
Loss: Loss:
name: CombinedLoss name: CombinedLoss
loss_config_list: loss_config_list:
......
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