diff --git a/timm/models/layers/helpers.py b/timm/models/layers/helpers.py index 371b3cf6..7a738d5c 100644 --- a/timm/models/layers/helpers.py +++ b/timm/models/layers/helpers.py @@ -3,13 +3,13 @@ Hacked together by / Copyright 2020 Ross Wightman """ from itertools import repeat -from torch._six import container_abcs +import collections.abc # From PyTorch internals def _ntuple(n): def parse(x): - if isinstance(x, container_abcs.Iterable): + if isinstance(x, collections.abc.Iterable): return x return tuple(repeat(x, n)) return parse