qat_dis.yaml 745 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
Global:
  input_name: x
  model_dir: EfficientNetB0_infer
  model_filename: inference.pdmodel
  params_filename: inference.pdiparams
  batch_size: 32
  data_dir: /ILSVRC2012

Distillation:
  alpha: 1.0
  loss: l2
  node:
  - softmax_1.tmp_0
Quantization:
  use_pact: true
  activation_bits: 8
  is_full_quantize: false
18
  activation_quantize_type: moving_average_abs_max
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
  weight_quantize_type: channel_wise_abs_max
  not_quant_pattern:
  - skip_quant
  quantize_op_types:
  - conv2d
  - depthwise_conv2d
  weight_bits: 8
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.7738