Skip to content

Commit c2fe0a2

Browse files
a-r-r-o-wrwightman
authored andcommitted
improvement: add typehints and docs to timm/models/mobilenetv3.py
1 parent d023154 commit c2fe0a2

File tree

2 files changed

+116
-67
lines changed

2 files changed

+116
-67
lines changed

timm/models/_typing.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
import functools
2+
import types
3+
from typing import Any, Dict, List, Tuple, Union
4+
5+
import torch.nn
6+
7+
8+
BlockArgs = List[List[Dict[str, Any]]]
9+
LayerType = Union[type, str, types.FunctionType, functools.partial, torch.nn.Module]
10+
PadType = Union[str, int, Tuple[int, int]]

0 commit comments

Comments
 (0)