mmagic.utils.typing 源代码

# Copyright (c) OpenMMLab. All rights reserved.
from typing import Callable, Dict, List, Sequence, Tuple, Union

from mmengine.config import ConfigDict
from mmengine.structures import BaseDataElement
from torch import Tensor

[文档]ForwardInputs = Tuple[Dict[str, Union[Tensor, str, int]], Tensor]
[文档]SampleList = Sequence[BaseDataElement]

[文档]NoiseVar = Union[Tensor, Callable, None]
[文档]LabelVar = Union[Tensor, Callable, List[int], None]

[文档]ConfigType = Union[ConfigDict, Dict]