hopefully finally fix torch/torchvision reqs?

nschaetti · Jun 10, 2024 · eb7c50d · eb7c50d
1 parent 7bae012
commit eb7c50d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -5,6 +5,7 @@ numpy==1.21
 scipy==1.10
 scikit-learn
 matplotlib
-torchvision==0.11
+torch==1.10
+torchvision
 networkx
 tqdm