hasibzunair commited on
Commit
21e19cf
1 Parent(s): 1f43f85
Files changed (2) hide show
  1. pipeline/timm_utils/tuple.py +2 -2
  2. requirements.txt +2 -2
pipeline/timm_utils/tuple.py CHANGED
@@ -3,8 +3,8 @@
3
  Hacked together by / Copyright 2020 Ross Wightman
4
  """
5
  from itertools import repeat
6
- from torch._six import container_abcs
7
-
8
 
9
  # From PyTorch internals
10
  def _ntuple(n):
 
3
  Hacked together by / Copyright 2020 Ross Wightman
4
  """
5
  from itertools import repeat
6
+ #from torch._six import container_abcs
7
+ import collections.abc as container_abcs
8
 
9
  # From PyTorch internals
10
  def _ntuple(n):
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- timm==0.4.12
2
  torch==1.13.0
3
  torchvision
4
- Pillow
 
 
 
1
  torch==1.13.0
2
  torchvision
3
+ Pillow
4
+ collections