config.yaml 677 字节
Newer Older
W
whs 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
version: 1.0
include: ["./unitest/configs/pruners.yaml", "./unitest/configs/pruners_0.yaml"]
pruners:
    pruner_1:
        class: 'RatioPruner'
        ratios:
            'conv1_1.w': 0.3
            'conv1_2.w': 0.4
            '*': 0.9
        group_dims:
            '*': [1, 2, 3]
        criterions:
            '*': 'l1-norm'
strategies:
    strategy_1:
        class: 'SensitivePruneStrategy'
        pruner: 'pruner_2'
        start_epoch: 0
        end_epoch: 10
        delta_rate: 0.20
        acc_loss_threshold: 0.2
        sensitivities:
            'conv1_1.w': 0.4

compress_pass:
    class: 'CompressPass'
    epoch: 100
    strategies:
        - strategy_1