diff --git a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml index 0553a65e448db21f196389f6abba7bf8808f0494..f3ad966488dbc2f6f7ca12033bc4a3d35e1b3bd7 100644 --- a/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml +++ b/configs/det/ch_PP-OCRv2/ch_PP-OCRv2_det_dml.yml @@ -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