qat_dis.yaml 757 字节
Newer Older
1
Global:
C
Chang Xu 已提交
2
  model_dir: save_quant_inception
3 4 5
  model_filename: inference.pdmodel
  params_filename: inference.pdiparams
  batch_size: 32
C
Chang Xu 已提交
6 7
  resize_size: 320
  img_size: 299
C
Chang Xu 已提交
8
  data_dir: ./ILSVRC2012
9 10
  
Distillation:
C
Chang Xu 已提交
11
  alpha: 1.0
12 13 14 15 16
  loss: l2
  node:
  - softmax_1.tmp_0
Quantization:
  is_full_quantize: false
C
Chang Xu 已提交
17
  activation_quantize_type: moving_average_abs_max
18 19 20 21 22 23
  weight_quantize_type: channel_wise_abs_max
  not_quant_pattern:
  - skip_quant
  quantize_op_types:
  - conv2d
  - depthwise_conv2d
24 25
  - matmul
  - matmul_v2
26
  weight_bits: 8
C
Chang Xu 已提交
27
  
28 29 30 31 32 33 34 35 36 37 38
TrainConfig:
  epochs: 1
  eval_iter: 500
  learning_rate: 
    type: CosineAnnealingDecay 
    learning_rate: 0.015
  optimizer_builder:
    optimizer:
      type: Momentum
    weight_decay: 0.00002
  origin_metric: 0.7914