From e6b72ca79e48ac0c6983e0db69c657e666e6d315 Mon Sep 17 00:00:00 2001 From: humu789 Date: Mon, 16 Jan 2023 17:24:50 +0800 Subject: [PATCH] fix import --- mmrazor/models/quantizers/academic_quantizer.py | 2 +- mmrazor/models/quantizers/native_quantizer.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/mmrazor/models/quantizers/academic_quantizer.py b/mmrazor/models/quantizers/academic_quantizer.py index edcb6b0c9..12e057ec9 100644 --- a/mmrazor/models/quantizers/academic_quantizer.py +++ b/mmrazor/models/quantizers/academic_quantizer.py @@ -3,7 +3,7 @@ import torch -from mmrazor.models.task_modules import build_graphmodule +from mmrazor.models.task_modules.tracer import build_graphmodule from mmrazor.models.utils import str2class from mmrazor.registry import MODELS from mmrazor.structures.quantization import BackendConfigs, QConfigHander diff --git a/mmrazor/models/quantizers/native_quantizer.py b/mmrazor/models/quantizers/native_quantizer.py index 8b416aeb0..5184adfb5 100644 --- a/mmrazor/models/quantizers/native_quantizer.py +++ b/mmrazor/models/quantizers/native_quantizer.py @@ -20,6 +20,7 @@ qat_modules = get_package_placeholder('torch>=1.13') from mmrazor import digit_version +from mmrazor.models.task_modules.tracer import build_graphmodule from mmrazor.models.task_modules.tracer.fx import ( del_fakequant_after_function, del_fakequant_after_method, del_fakequant_after_module, del_fakequant_after_op,