diff --git a/example/auto_compression/image_classification/configs/EfficientNetB0/qat_dis.yaml b/example/auto_compression/image_classification/configs/EfficientNetB0/qat_dis.yaml index c74e03211d320214fb51e21d110feaa1a5721171..a1fe00d7193e7c97a94c3047d7a1527776d4bd65 100644 --- a/example/auto_compression/image_classification/configs/EfficientNetB0/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/EfficientNetB0/qat_dis.yaml @@ -24,6 +24,8 @@ Quantization: quantize_op_types: - conv2d - depthwise_conv2d + - matmul + - matmul_v2 weight_bits: 8 TrainConfig: 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 4d6bb699921c3300a470831a0d259bf28e81007a..1b96e09a322ab26fb07f94c776653407445f1b2d 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 @@ -22,6 +22,8 @@ Quantization: quantize_op_types: - conv2d - depthwise_conv2d + - matmul + - matmul_v2 weight_bits: 8 TrainConfig: epochs: 1 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 a4615c201ff2c2fa6552507855dfb9b01190c062..e3f3384dda3987013fc752fa3071bb2807b5dca9 100644 --- a/example/auto_compression/image_classification/configs/InceptionV3/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/InceptionV3/qat_dis.yaml @@ -22,6 +22,8 @@ Quantization: quantize_op_types: - conv2d - depthwise_conv2d + - matmul + - matmul_v2 weight_bits: 8 TrainConfig: 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 b0fff1454ba3b8254dbfff4d20a22af870d807d1..e83c3fee2a395c1627377db16984403a7e66a5d5 100644 --- a/example/auto_compression/image_classification/configs/MobileNetV1/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/MobileNetV1/qat_dis.yaml @@ -22,6 +22,8 @@ Quantization: quantize_op_types: - conv2d - depthwise_conv2d + - matmul + - matmul_v2 weight_bits: 8 TrainConfig: epochs: 1 diff --git a/example/auto_compression/image_classification/configs/MobileNetV3_large_x1_0/qat_dis.yaml b/example/auto_compression/image_classification/configs/MobileNetV3_large_x1_0/qat_dis.yaml index f4780e0b3b6882f34782e63be9f2cba9562346e7..7f5eceb4f4837083e262ae5517720deff40e5ac2 100644 --- a/example/auto_compression/image_classification/configs/MobileNetV3_large_x1_0/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/MobileNetV3_large_x1_0/qat_dis.yaml @@ -23,6 +23,7 @@ Quantization: - conv2d - depthwise_conv2d - matmul + - matmul_v2 weight_bits: 8 TrainConfig: diff --git a/example/auto_compression/image_classification/configs/PPHGNet_tiny/qat_dis.yaml b/example/auto_compression/image_classification/configs/PPHGNet_tiny/qat_dis.yaml index fb1535e7bc1abeeecb641b8339b16fa34dc7b0f6..661c1bca6c4dc273997e43fcfc0bc979196f19d9 100644 --- a/example/auto_compression/image_classification/configs/PPHGNet_tiny/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/PPHGNet_tiny/qat_dis.yaml @@ -24,6 +24,8 @@ Quantization: quantize_op_types: - conv2d - depthwise_conv2d + - matmul + - matmul_v2 weight_bits: 8 TrainConfig: diff --git a/example/auto_compression/image_classification/configs/PPLCNetV2_base/qat_dis.yaml b/example/auto_compression/image_classification/configs/PPLCNetV2_base/qat_dis.yaml index 4be08fe371a75f0ed116b7172129d747d1a1bf69..018e48b4ee44d34bfc9df9b7a3b09f62ae9481dd 100644 --- a/example/auto_compression/image_classification/configs/PPLCNetV2_base/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/PPLCNetV2_base/qat_dis.yaml @@ -24,6 +24,8 @@ Quantization: quantize_op_types: - conv2d - depthwise_conv2d + - matmul + - matmul_v2 weight_bits: 8 TrainConfig: 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 394455b3bbe62d794730ca9827f0b4d0b1ea3310..8c823f25a3c125273e82f208d1fa226fd853b017 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 @@ -22,6 +22,8 @@ Quantization: quantize_op_types: - conv2d - depthwise_conv2d + - matmul + - matmul_v2 weight_bits: 8 TrainConfig: epochs: 1 diff --git a/example/auto_compression/image_classification/configs/ResNet50_vd/qat_dis.yaml b/example/auto_compression/image_classification/configs/ResNet50_vd/qat_dis.yaml index 4cc375add192b9a01f70ceb6c2b0e7f7c49288ca..e94d509bf1d7716059b8e035a594eba2d6d9bc7c 100644 --- a/example/auto_compression/image_classification/configs/ResNet50_vd/qat_dis.yaml +++ b/example/auto_compression/image_classification/configs/ResNet50_vd/qat_dis.yaml @@ -24,6 +24,8 @@ Quantization: quantize_op_types: - conv2d - depthwise_conv2d + - matmul + - matmul_v2 weight_bits: 8 TrainConfig: 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 b0d3ffb44695eb036cf001e0469e92f7187717fd..4792f490cab34ff84da79217aaaef44166f27e05 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 @@ -22,6 +22,8 @@ Quantization: quantize_op_types: - conv2d - depthwise_conv2d + - matmul + - matmul_v2 weight_bits: 8 TrainConfig: epochs: 1 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 086855377784f68e5f8f6822aa38fcbf797ceeb0..01a0b153e084840a3078925708ef725c528e6c4b 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 @@ -21,6 +21,8 @@ Quantization: quantize_op_types: - conv2d - depthwise_conv2d + - matmul + - matmul_v2 weight_bits: 8 TrainConfig: epochs: 1 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 d07e734aa5cd9936986ee6cd4c40e66cf62a5449..f1fc3a1f46158f86120eb90ca45feca338b753fa 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 @@ -22,6 +22,8 @@ Quantization: quantize_op_types: - conv2d - depthwise_conv2d + - matmul + - matmul_v2 weight_bits: 8 TrainConfig: epochs: 1