fix load issue (#194)

Signed-off-by: changwa1 <[email protected]>
Signed-off-by: Lv, Liang1 <[email protected]>

Author: changwangss

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__

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]