hasibzunair commited on
Commit
1f43f85
·
1 Parent(s): e834bbd
Files changed (2) hide show
  1. pipeline/timm_utils/tuple.py +2 -2
  2. requirements.txt +1 -0
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
- import container.abc as container_abcs
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
+
8
 
9
  # From PyTorch internals
10
  def _ntuple(n):
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  torch==1.13.0
2
  torchvision
3
  Pillow
 
1
+ timm==0.4.12
2
  torch==1.13.0
3
  torchvision
4
  Pillow