fix import

open-mmlab · Jan 16, 2023 · e6b72ca · e6b72ca
1 parent 7f3a78c
commit e6b72ca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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,