mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Allow ntuple to be used with string values

Browse Source
This commit is contained in:
Ross Wightman 2022-07-28 16:18:18 -07:00
parent 7430a85d07
commit 8ad4bdfa06
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/layers/helpers.py
View File

@ -9,7 +9,7 @@ import collections.abc
# From PyTorch internals # From PyTorch internals
def _ntuple(n): def _ntuple(n):
def parse(x): def parse(x):
if isinstance(x, collections.abc.Iterable): if isinstance(x, collections.abc.Iterable) and not isinstance(x, str):
return x return x
return tuple(repeat(x, n)) return tuple(repeat(x, n))
return parse return parse