Skip to content

Commit 41fa84a

Browse files
authored
[Fix] remove torch dependencies in build_function.py (#1632)
1 parent 698782f commit 41fa84a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mmengine/registry/build_functions.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,6 @@
33
import logging
44
from typing import TYPE_CHECKING, Any, Optional, Union
55

6-
import torch
7-
86
from mmengine.config import Config, ConfigDict
97
from mmengine.utils import ManagerMixin, digit_version
108
from .registry import Registry
@@ -238,6 +236,8 @@ def build_optimizer_from_cfg(
238236
cfg: Union[dict, ConfigDict, Config],
239237
registry: Registry,
240238
default_args: Optional[Union[dict, ConfigDict, Config]] = None) -> Any:
239+
import torch
240+
241241
from ..logging import print_log
242242
if 'type' in cfg \
243243
and 'Adafactor' == cfg['type'] \

0 commit comments

Comments
 (0)