From 21b5886500aca54fd68266582fcea69ffa1bf089 Mon Sep 17 00:00:00 2001 From: Bai Yifan Date: Wed, 6 Nov 2019 10:47:18 +0800 Subject: [PATCH] fix distillation demo config (#3880) --- .../configs/mobilenetv1_resnet50_distillation.yaml | 7 +------ .../configs/resnet34_resnet50_distillation.yaml | 7 +------ 2 files changed, 2 insertions(+), 12 deletions(-) diff --git a/PaddleSlim/classification/distillation/configs/mobilenetv1_resnet50_distillation.yaml b/PaddleSlim/classification/distillation/configs/mobilenetv1_resnet50_distillation.yaml index 03c0d6ca..750cbae2 100644 --- a/PaddleSlim/classification/distillation/configs/mobilenetv1_resnet50_distillation.yaml +++ b/PaddleSlim/classification/distillation/configs/mobilenetv1_resnet50_distillation.yaml @@ -1,10 +1,5 @@ version: 1.0 distillers: - fsp_distiller: - class: 'FSPDistiller' - teacher_pairs: [['res50_res2a_branch2a.conv2d.output.1.tmp_0', 'res50_res3a_branch2a.conv2d.output.1.tmp_0']] - student_pairs: [['depthwise_conv2d_1.tmp_0', 'depthwise_conv2d_2.tmp_0']] - distillation_loss_weight: 1 l2_distiller: class: 'L2Distiller' teacher_feature_map: 'res50_fc_0.tmp_0' @@ -13,7 +8,7 @@ distillers: strategies: distillation_strategy: class: 'DistillationStrategy' - distillers: ['fsp_distiller', 'l2_distiller'] + distillers: ['l2_distiller'] start_epoch: 0 end_epoch: 130 compressor: diff --git a/PaddleSlim/classification/distillation/configs/resnet34_resnet50_distillation.yaml b/PaddleSlim/classification/distillation/configs/resnet34_resnet50_distillation.yaml index 4d4546e0..e19dc0e9 100644 --- a/PaddleSlim/classification/distillation/configs/resnet34_resnet50_distillation.yaml +++ b/PaddleSlim/classification/distillation/configs/resnet34_resnet50_distillation.yaml @@ -1,10 +1,5 @@ version: 1.0 distillers: - fsp_distiller: - class: 'FSPDistiller' - teacher_pairs: [['res50_res2a_branch2a.conv2d.output.1.tmp_0', 'res50_res2b_branch2a.conv2d.output.1.tmp_0'], ['res50_res4b_branch2a.conv2d.output.1.tmp_0', 'res50_res4c_branch2a.conv2d.output.1.tmp_0']] - student_pairs: [['res34_res2a_branch2a.conv2d.output.1.tmp_0', 'res34_res2a_branch1.conv2d.output.1.tmp_0'], ['res34_res4b_branch2a.conv2d.output.1.tmp_0', 'res34_res4c_branch2a.conv2d.output.1.tmp_0']] - distillation_loss_weight: 1 l2_distiller: class: 'L2Distiller' teacher_feature_map: 'res50_fc_0.tmp_0' @@ -13,7 +8,7 @@ distillers: strategies: distillation_strategy: class: 'DistillationStrategy' - distillers: ['fsp_distiller', 'l2_distiller'] + distillers: ['l2_distiller'] start_epoch: 0 end_epoch: 130 compressor: -- GitLab