balthou commited on
Commit
4cc6d89
·
1 Parent(s): 935d706

add torchvision to reqs

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -0
requirements.txt CHANGED
@@ -1,7 +1,10 @@
1
  interactive-pipe>=0.7.8
2
  torch>=2.0.0
 
 
3
  tqdm
4
  numba
5
  scikit-image
6
 
7
 
 
 
1
  interactive-pipe>=0.7.8
2
  torch>=2.0.0
3
+ torchvision
4
+ torchmetrics
5
  tqdm
6
  numba
7
  scikit-image
8
 
9
 
10
+