config.yaml

version: 1.0
include: ["./configs/pruners.yaml", "./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