mirror of
https://github.com/open-mmlab/mmengine.git
synced 2025-06-03 21:54:44 +08:00
[Fix] remove torch dependencies in build_function.py
(#1632)
This commit is contained in:
parent
698782f920
commit
41fa84a9a9
@ -3,8 +3,6 @@ import inspect
|
|||||||
import logging
|
import logging
|
||||||
from typing import TYPE_CHECKING, Any, Optional, Union
|
from typing import TYPE_CHECKING, Any, Optional, Union
|
||||||
|
|
||||||
import torch
|
|
||||||
|
|
||||||
from mmengine.config import Config, ConfigDict
|
from mmengine.config import Config, ConfigDict
|
||||||
from mmengine.utils import ManagerMixin, digit_version
|
from mmengine.utils import ManagerMixin, digit_version
|
||||||
from .registry import Registry
|
from .registry import Registry
|
||||||
@ -238,6 +236,8 @@ def build_optimizer_from_cfg(
|
|||||||
cfg: Union[dict, ConfigDict, Config],
|
cfg: Union[dict, ConfigDict, Config],
|
||||||
registry: Registry,
|
registry: Registry,
|
||||||
default_args: Optional[Union[dict, ConfigDict, Config]] = None) -> Any:
|
default_args: Optional[Union[dict, ConfigDict, Config]] = None) -> Any:
|
||||||
|
import torch
|
||||||
|
|
||||||
from ..logging import print_log
|
from ..logging import print_log
|
||||||
if 'type' in cfg \
|
if 'type' in cfg \
|
||||||
and 'Adafactor' == cfg['type'] \
|
and 'Adafactor' == cfg['type'] \
|
||||||
|
Loading…
x
Reference in New Issue
Block a user