Merge pull request #421 from aboutmedicine/master

Update collections.abc import to avoid pytorch 1.8 issue. Fix #420

Author: rwightman

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