From 3b9821c8146c838ccc976225023ec1be90ce176f Mon Sep 17 00:00:00 2001 From: changwa1 Date: Wed, 30 Nov 2022 17:39:11 +0800 Subject: [PATCH] fix load issue Signed-off-by: changwa1 --- .../torchvision_models/quantization/ptq/cpu/ipex/main.py | 1 + .../ssd_resnet34/quantization/ptq/ipex/ssd_r34.py | 1 + 2 files changed, 2 insertions(+) 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]