mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
11 lines
265 B
Python
11 lines
265 B
Python
import functools
|
|
import types
|
|
from typing import Any, Dict, List, Tuple, Union
|
|
|
|
import torch.nn
|
|
|
|
|
|
BlockArgs = List[List[Dict[str, Any]]]
|
|
LayerType = Union[type, str, types.FunctionType, functools.partial, torch.nn.Module]
|
|
PadType = Union[str, int, Tuple[int, int]]
|