2023-10-30 05:07:50 +08:00
|
|
|
from typing import Callable, Tuple, Type, Union
|
2023-10-19 21:20:38 +08:00
|
|
|
|
2023-10-30 05:07:50 +08:00
|
|
|
import torch
|
2023-10-19 21:20:38 +08:00
|
|
|
|
|
|
|
|
|
2023-10-30 05:07:50 +08:00
|
|
|
LayerType = Union[str, Callable, Type[torch.nn.Module]]
|
2023-10-19 21:20:38 +08:00
|
|
|
PadType = Union[str, int, Tuple[int, int]]
|
