mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
29f034cf93
* Fix links in readme * Fix links in readme * Update PPOCRv2/v3 examples * Update auto compression configs
40 lines
918 B
YAML
40 lines
918 B
YAML
Global:
|
|
model_dir: ./yolov6s.onnx
|
|
format: 'onnx'
|
|
model_filename: model.pdmodel
|
|
params_filename: model.pdiparams
|
|
qat_image_path: ./COCO_train_320
|
|
ptq_image_path: ./COCO_val_320
|
|
input_list: ['x2paddle_image_arrays']
|
|
qat_preprocess: yolo_image_preprocess
|
|
ptq_preprocess: yolo_image_preprocess
|
|
qat_batch_size: 1
|
|
|
|
Distillation:
|
|
alpha: 1.0
|
|
loss: soft_label
|
|
|
|
QuantAware:
|
|
onnx_format: true
|
|
activation_quantize_type: 'moving_average_abs_max'
|
|
quantize_op_types:
|
|
- conv2d
|
|
- depthwise_conv2d
|
|
- conv2d_transpose
|
|
|
|
|
|
PTQ:
|
|
calibration_method: 'avg' # option: avg, abs_max, hist, KL, mse
|
|
skip_tensor_list: ['conv2d_2.w_0', 'conv2d_15.w_0', 'conv2d_46.w_0', 'conv2d_11.w_0', 'conv2d_49.w_0']
|
|
|
|
TrainConfig:
|
|
train_iter: 8000
|
|
learning_rate:
|
|
type: CosineAnnealingDecay
|
|
learning_rate: 0.00003
|
|
T_max: 8000
|
|
optimizer_builder:
|
|
optimizer:
|
|
type: SGD
|
|
weight_decay: 0.00004
|