diff --git a/example/auto_compression/semantic_segmentation/configs/hrnet/hrnet_qat.yaml b/example/auto_compression/semantic_segmentation/configs/hrnet/hrnet_qat.yaml index 36d3fde4a8f00eabf06d2800a279b7b42a77d279..1eec456e267427d7a22545c63eeae82b5c5bc68b 100644 --- a/example/auto_compression/semantic_segmentation/configs/hrnet/hrnet_qat.yaml +++ b/example/auto_compression/semantic_segmentation/configs/hrnet/hrnet_qat.yaml @@ -11,6 +11,7 @@ Distillation: - conv2d_613.tmp_1 Quantization: + onnx_format: True quantize_op_types: - conv2d - depthwise_conv2d diff --git a/example/auto_compression/semantic_segmentation/configs/pp_humanseg/pp_humanseg_qat.yaml b/example/auto_compression/semantic_segmentation/configs/pp_humanseg/pp_humanseg_qat.yaml index 8a917f985b4844b2028c85e8e39eb3603302ee26..8893dc35c730c57380394696bc4b0147266224dc 100644 --- a/example/auto_compression/semantic_segmentation/configs/pp_humanseg/pp_humanseg_qat.yaml +++ b/example/auto_compression/semantic_segmentation/configs/pp_humanseg/pp_humanseg_qat.yaml @@ -11,6 +11,7 @@ Distillation: - batch_norm_47.tmp_2 Quantization: + onnx_format: True quantize_op_types: - conv2d - depthwise_conv2d diff --git a/example/auto_compression/semantic_segmentation/configs/pp_liteseg/pp_liteseg_qat.yaml b/example/auto_compression/semantic_segmentation/configs/pp_liteseg/pp_liteseg_qat.yaml index 67ee9d69073a94a54ef8801d055ccaef29782727..12eea7e26b79272a41fabd5a9daa7282fcf77662 100644 --- a/example/auto_compression/semantic_segmentation/configs/pp_liteseg/pp_liteseg_qat.yaml +++ b/example/auto_compression/semantic_segmentation/configs/pp_liteseg/pp_liteseg_qat.yaml @@ -11,6 +11,7 @@ Distillation: - conv2d_95.tmp_0 Quantization: + onnx_format: True quantize_op_types: - conv2d - depthwise_conv2d diff --git a/example/auto_compression/semantic_segmentation/configs/unet/unet_qat.yaml b/example/auto_compression/semantic_segmentation/configs/unet/unet_qat.yaml index 7ca808b2febeb95fa387e6ee7c172a9a98bc5245..ff055e2b02361142a951a2c5bbb39bba8a8fa4ba 100644 --- a/example/auto_compression/semantic_segmentation/configs/unet/unet_qat.yaml +++ b/example/auto_compression/semantic_segmentation/configs/unet/unet_qat.yaml @@ -11,6 +11,7 @@ Distillation: - conv2d_37.tmp_1 Quantization: + onnx_format: True quantize_op_types: - conv2d - depthwise_conv2d