fix bug under pt13

open-mmlab · Jan 13, 2023 · 296930b · 296930b
1 parent b427d81
commit 296930b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_runners/test_quantization_loop.py b/tests/test_runners/test_quantization_loop.py
@@ -15,7 +15,6 @@
 from mmengine.optim import OptimWrapper
 from mmengine.registry import DATASETS, HOOKS, METRICS, MODELS, OPTIM_WRAPPERS
 from mmengine.runner import Runner
-from torch.ao.quantization.qconfig_mapping import get_default_qconfig_mapping
 from torch.nn.intrinsic.qat import ConvBnReLU2d
 from torch.utils.data import Dataset
 
@@ -26,13 +25,16 @@
     from torch.ao.quantization import QConfigMapping
     from torch.ao.quantization.fake_quantize import FakeQuantizeBase
     from torch.ao.quantization.fx import prepare
+    from torch.ao.quantization.qconfig_mapping import \
+        get_default_qconfig_mapping
     from torch.ao.quantization.quantize_fx import _fuse_fx
 except ImportError:
     from mmrazor.utils import get_placeholder
     QConfigMapping = get_placeholder('torch>=1.13')
     FakeQuantizeBase = get_placeholder('torch>=1.13')
     prepare = get_placeholder('torch>=1.13')
     _fuse_fx = get_placeholder('torch>=1.13')
+    get_default_qconfig_mapping = get_placeholder('torch>=1.13')
 
 from mmrazor import digit_version