Try torch 1.10

qubvel-org · Oct 7, 2024 · 963e6ad · 963e6ad
1 parent 2d1a12b
commit 963e6ad
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -24,8 +24,8 @@ dependencies = [
     'pretrainedmodels>=0.7.1',
     'six>=1.5',
     'timm>=0.9',
-    'torch>=1.9',
-    'torchvision>=0.10',
+    'torch>=1.10',
+    'torchvision>=0.11',
     'tqdm>=4.42.1',
 ]
 dynamic = ['version']

diff --git a/requirements/minimum.old b/requirements/minimum.old
@@ -5,6 +5,6 @@ pillow==8.0.0
 pretrainedmodels==0.7.1
 six==1.5.0
 timm==0.9.0
-torch==1.9.0
-torchvision==0.10.0
+torch==1.10.0
+torchvision==0.11.0
 tqdm==4.42.1