From 41c4780a69041ecb257b7c4d473c49e9fd10c4ea Mon Sep 17 00:00:00 2001 From: iamWHTWD <28421344+iamWHTWD@users.noreply.github.com> Date: Tue, 7 Feb 2023 17:36:37 +0800 Subject: [PATCH] set 'onnx_format: True' with ACT case (#1649) * set onnx_format: True * set onnx_format: True --- .../image_classification/configs/GhostNet_x1_0/qat_dis.yaml | 1 + .../image_classification/configs/InceptionV3/qat_dis.yaml | 1 + .../image_classification/configs/MobileNetV1/qat_dis.yaml | 1 + .../image_classification/configs/PPLCNet_x1_0/qat_dis.yaml | 1 + .../image_classification/configs/ShuffleNetV2_x1_0/qat_dis.yaml | 1 + .../image_classification/configs/SqueezeNet1_0/qat_dis.yaml | 1 + .../configs/SwinTransformer_base_patch4_window7_224/qat_dis.yaml | 1 + 7 files changed, 7 insertions(+) diff --git a/example/auto_compression/image_classification/configs/GhostNet_x1_0/qat_dis.yaml b/example/auto_compression/image_classification/configs/GhostNet_x1_0/qat_dis.yaml index 0e91d4c0..4b5448a5 100644 --- a/example/auto_compression/image_classification/configs/GhostNet_x1_0/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/GhostNet_x1_0/qat_dis.yaml @@ -14,6 +14,7 @@ QuantAware: use_pact: true activation_bits: 8 is_full_quantize: false + onnx_format: True activation_quantize_type: moving_average_abs_max weight_quantize_type: channel_wise_abs_max not_quant_pattern: diff --git a/example/auto_compression/image_classification/configs/InceptionV3/qat_dis.yaml b/example/auto_compression/image_classification/configs/InceptionV3/qat_dis.yaml index 3b1e4084..64ec6a3b 100644 --- a/example/auto_compression/image_classification/configs/InceptionV3/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/InceptionV3/qat_dis.yaml @@ -14,6 +14,7 @@ Distillation: - softmax_1.tmp_0 QuantAware: is_full_quantize: false + onnx_format: True activation_quantize_type: moving_average_abs_max weight_quantize_type: channel_wise_abs_max not_quant_pattern: diff --git a/example/auto_compression/image_classification/configs/MobileNetV1/qat_dis.yaml b/example/auto_compression/image_classification/configs/MobileNetV1/qat_dis.yaml index 8f74d745..6ad0174e 100644 --- a/example/auto_compression/image_classification/configs/MobileNetV1/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/MobileNetV1/qat_dis.yaml @@ -14,6 +14,7 @@ QuantAware: use_pact: true activation_bits: 8 is_full_quantize: false + onnx_format: True activation_quantize_type: moving_average_abs_max weight_quantize_type: channel_wise_abs_max not_quant_pattern: diff --git a/example/auto_compression/image_classification/configs/PPLCNet_x1_0/qat_dis.yaml b/example/auto_compression/image_classification/configs/PPLCNet_x1_0/qat_dis.yaml index f0e67260..56e1a992 100644 --- a/example/auto_compression/image_classification/configs/PPLCNet_x1_0/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/PPLCNet_x1_0/qat_dis.yaml @@ -14,6 +14,7 @@ QuantAware: use_pact: true activation_bits: 8 is_full_quantize: false + onnx_format: True activation_quantize_type: moving_average_abs_max weight_quantize_type: channel_wise_abs_max not_quant_pattern: diff --git a/example/auto_compression/image_classification/configs/ShuffleNetV2_x1_0/qat_dis.yaml b/example/auto_compression/image_classification/configs/ShuffleNetV2_x1_0/qat_dis.yaml index 31c618e4..79b16b39 100644 --- a/example/auto_compression/image_classification/configs/ShuffleNetV2_x1_0/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/ShuffleNetV2_x1_0/qat_dis.yaml @@ -14,6 +14,7 @@ QuantAware: use_pact: true activation_bits: 8 is_full_quantize: false + onnx_format: True activation_quantize_type: moving_average_abs_max weight_quantize_type: channel_wise_abs_max not_quant_pattern: diff --git a/example/auto_compression/image_classification/configs/SqueezeNet1_0/qat_dis.yaml b/example/auto_compression/image_classification/configs/SqueezeNet1_0/qat_dis.yaml index 4b9964af..e4b4d8f6 100644 --- a/example/auto_compression/image_classification/configs/SqueezeNet1_0/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/SqueezeNet1_0/qat_dis.yaml @@ -13,6 +13,7 @@ Distillation: QuantAware: activation_bits: 8 is_full_quantize: false + onnx_format: True activation_quantize_type: moving_average_abs_max weight_quantize_type: channel_wise_abs_max not_quant_pattern: diff --git a/example/auto_compression/image_classification/configs/SwinTransformer_base_patch4_window7_224/qat_dis.yaml b/example/auto_compression/image_classification/configs/SwinTransformer_base_patch4_window7_224/qat_dis.yaml index 99f61b77..9d7375cc 100644 --- a/example/auto_compression/image_classification/configs/SwinTransformer_base_patch4_window7_224/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/SwinTransformer_base_patch4_window7_224/qat_dis.yaml @@ -14,6 +14,7 @@ QuantAware: use_pact: true activation_bits: 8 is_full_quantize: false + onnx_format: True activation_quantize_type: moving_average_abs_max weight_quantize_type: channel_wise_abs_max not_quant_pattern: -- GitLab