No torchvision version

NKI-AI · Oct 24, 2024 · 5c2fea3 · 5c2fea3
1 parent 9a45371
commit 5c2fea3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -59,7 +59,7 @@ def finalize_options(self):
         "h5py==3.11.0",
         "omegaconf==2.3.0",
         "torch>=2.2.0",
-        "torchvision==0.18.0",
+        "torchvision",
         "scikit-image>=0.19.0",
         "scikit-learn>=1.0.1",
         "tensorboard>=2.7.0",