diff --git a/examples/pytorch/image_recognition/torchvision_models/quantization/ptq/cpu/ipex/main.py b/examples/pytorch/image_recognition/torchvision_models/quantization/ptq/cpu/ipex/main.py index 7186718ab61..a5ce5a9885c 100644 --- a/examples/pytorch/image_recognition/torchvision_models/quantization/ptq/cpu/ipex/main.py +++ b/examples/pytorch/image_recognition/torchvision_models/quantization/ptq/cpu/ipex/main.py @@ -24,6 +24,7 @@ import torchvision.models as models from neural_compressor.adaptor.pytorch import get_torch_version from packaging.version import Version +import intel_extension_for_pytorch model_names = sorted(name for name in models.__dict__ diff --git a/examples/pytorch/object_detection/ssd_resnet34/quantization/ptq/ipex/ssd_r34.py b/examples/pytorch/object_detection/ssd_resnet34/quantization/ptq/ipex/ssd_r34.py index 5edbe2580ad..4e2db16cb99 100644 --- a/examples/pytorch/object_detection/ssd_resnet34/quantization/ptq/ipex/ssd_r34.py +++ b/examples/pytorch/object_detection/ssd_resnet34/quantization/ptq/ipex/ssd_r34.py @@ -24,6 +24,7 @@ from base_model import ResNet34 from typing import List +import intel_extension_for_pytorch Vector = List[torch.Tensor]