rtdetr_r101vd_qat_dis.yaml 620 字节
Newer Older
X
xiaoluomi 已提交
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 30 31 32

Global:
  reader_config: configs/rtdetr_reader.yml
  include_nms: True
  Evaluation: True
  model_dir: ./rtdetr_hgnetv2_x_6x_coco/
  model_filename: model.pdmodel
  params_filename: model.pdiparams

Distillation:
  alpha: 1.0
  loss: soft_label

QuantAware:
  onnx_format: true
  activation_quantize_type: 'moving_average_abs_max'
  quantize_op_types:
  - conv2d
  - depthwise_conv2d
  - matmul_v2

TrainConfig:
  train_iter: 200
  eval_iter: 50
  learning_rate:  
    type: CosineAnnealingDecay
    learning_rate: 0.00003
    T_max: 10000
  optimizer_builder:
    optimizer: 
      type: SGD
    weight_decay: 4.0e-05