deeplabv3_qat.yaml 486 字节
Newer Older
Z
zhouzj 已提交
1
Global:
Z
zhouzj 已提交
2
  reader_config: configs/dataset/cityscapes_1024x512_scale1.0.yml
C
ceci3 已提交
3 4 5
  model_dir: ./RES-paddle2-Deeplabv3-ResNet50
  model_filename: model
  params_filename: params
Z
zhouzj 已提交
6
  batch_size: 4
Z
zhouzj 已提交
7

Z
zhouzj 已提交
8
Distillation:
C
ceci3 已提交
9 10 11
  alpha: 1.0
  loss: l2
  node:
12
  - conv2d_123.tmp_1
C
ceci3 已提交
13

Z
zhouzj 已提交
14
Quantization:
Z
zhouzj 已提交
15
  onnx_format: True
Z
zhouzj 已提交
16 17 18
  quantize_op_types:
  - conv2d
  - depthwise_conv2d
C
ceci3 已提交
19

Z
zhouzj 已提交
20
TrainConfig:
Z
zhouzj 已提交
21
  epochs: 1
22 23
  eval_iter: 360
  learning_rate: 0.0001
C
ceci3 已提交
24 25 26
  optimizer_builder:
    optimizer: 
      type: SGD
Z
zhouzj 已提交
27
    weight_decay: 0.0005