[Fix] remove torch dependencies in build_function.py (#1632)

open-mmlab · Feb 18, 2025 · 41fa84a · 41fa84a
1 parent 698782f
commit 41fa84a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmengine/registry/build_functions.py b/mmengine/registry/build_functions.py
@@ -3,8 +3,6 @@
 import logging
 from typing import TYPE_CHECKING, Any, Optional, Union
 
-import torch
-
 from mmengine.config import Config, ConfigDict
 from mmengine.utils import ManagerMixin, digit_version
 from .registry import Registry
@@ -238,6 +236,8 @@ def build_optimizer_from_cfg(
         cfg: Union[dict, ConfigDict, Config],
         registry: Registry,
         default_args: Optional[Union[dict, ConfigDict, Config]] = None) -> Any:
+    import torch
+
     from ..logging import print_log
     if 'type' in cfg \
             and 'Adafactor' == cfg['type'] \