mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix type annotation for torchscript

swin_maxvit_resize
Ross Wightman 2023-08-04 23:03:52 -07:00
parent 7790ea709b
commit 7c0f492dbb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/swin_transformer.py
View File

@ -41,7 +41,7 @@ _int_or_tuple_2_t = Union[int, Tuple[int, int]]
def window_partition( def window_partition(
x: torch.Tensor, x: torch.Tensor,
window_size: Tuple[int, int], window_size: Tuple[int, int],
) -> Tuple[torch.Tensor, Tuple[int, int]]: ) -> torch.Tensor:
""" """
Partition into non-overlapping windows with padding if needed. Partition into non-overlapping windows with padding if needed.
Args: Args: