mobilenetv1_resnet50_distillation.yaml 467 字节
Newer Older
1 2 3 4
version: 1.0
distillers:
    l2_distiller:
        class: 'L2Distiller'
W
whs 已提交
5
        teacher_feature_map: 'res_fc.tmp_0'
6 7 8 9 10
        student_feature_map: 'fc_0.tmp_0'
        distillation_loss_weight: 1
strategies:
    distillation_strategy:
        class: 'DistillationStrategy'
11
        distillers: ['l2_distiller']
12 13 14 15 16 17 18
        start_epoch: 0
        end_epoch: 130
compressor:
    epoch: 130
    checkpoint_path: './checkpoints/'
    strategies:
        - distillation_strategy