fix bug

Signed-off-by: Xin He <[email protected]>

Author: xin3he

examples/pytorch/image_recognition/torchvision_models/quantization/ptq/cpu/fx/main.py

@@ -181,6 +181,7 @@ def main():
             from neural_compressor.config import Torch2ONNXConfig
             fp32_onnx_config = Torch2ONNXConfig(
                 dtype="fp32",
+                opset_version=14,
                 example_inputs=torch.randn(1, 3, 224, 224),
                 input_names=['input'],
                 output_names=['output'],

examples/pytorch/image_recognition/torchvision_models/quantization/qat/fx/main.py

@@ -206,6 +206,7 @@ def training_func_for_nc(model):
             from neural_compressor.config import Torch2ONNXConfig
             fp32_onnx_config = Torch2ONNXConfig(
                 dtype="fp32",
+                opset_version=14,
                 example_inputs=torch.randn(1, 3, 224, 224),
                 input_names=['input'],
                 output_names=['output'],

examples/pytorch/nlp/huggingface_models/text-classification/quantization/ptq_static/fx/run_glue.py

@@ -519,7 +519,6 @@ def eval_func(model):
             fp32_onnx_config = Torch2ONNXConfig(
                 dtype="fp32",
                 opset_version=14,
-                quant_format="QDQ",
                 example_inputs=tuple(input.values()),
                 input_names=list(input.keys()),
                 output_names=['labels'],