upgrade to support pytorch2.0

youqingxiaozhua · Mar 31, 2023 · 6c7b576 · 6c7b576
1 parent f3b392b
commit 6c7b576
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmcls/models/vit/layers.py b/mmcls/models/vit/layers.py
@@ -223,10 +223,10 @@ def trunc_normal_(tensor, mean=0., std=1., a=-2., b=2.):
 ###
 # From PyTorch internals
 from itertools import repeat
-from torch._six import container_abcs
+import collections.abc
 def _ntuple(n) :
     def parse(x) -> Tuple[int]:
-        if isinstance(x, container_abcs.Iterable):
+        if isinstance(x, collections.abc.Iterable):
             return x
         return tuple(repeat(x, n))
     return parse