yolov3_mobilenet_v1_slim.yaml 605 字节
Newer Older
1 2 3 4 5
version: 1.0
strategies:
    quantization_strategy:
        class: 'QuantizationStrategy'
        start_epoch: 0
6
        end_epoch: 4 
7 8 9 10 11 12 13 14 15
        float_model_save_path: './output/yolov3/float'
        int8_model_save_path: './output/yolov3/int8'
        weight_bits: 8
        activation_bits: 8
        weight_quantize_type: 'abs_max'
        activation_quantize_type: 'moving_average_abs_max'
        save_in_nodes: ['image', 'im_size']
        save_out_nodes: ['multiclass_nms_0.tmp_0']
compressor:
16
    epoch: 5
17 18 19
    checkpoint_path: './checkpoints/yolov3/'
    strategies:
        - quantization_strategy