mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Mashiro 2025-02-18 16:38:51 +08:00 committed by GitHub
parent 698782f920
commit 41fa84a9a9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmengine/registry/build_functions.py
View File

@ -3,8 +3,6 @@ import inspect
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'] \