pytorch 1.7 -> 1.8

danielgafni · danielgafni · commit 4f0ebfd944b8 · 2021-04-17T19:07:35.000-04:00
diff --git a/requirements.txt b/requirements.txt
@@ -2,8 +2,8 @@ foolbox
 scipy>=1.1.0
 numpy>=1.14.2
 cython>=0.28.5
-torch==1.7.1
-torchvision==0.8.2
+torch==1.8.1
+torchvision==0.9.1
 tqdm>=4.19.9
 setuptools>=39.0.1
 matplotlib>=2.1.0
diff --git a/setup.py b/setup.py
@@ -17,8 +17,8 @@
     zip_safe=False,
     install_requires=[
         "numpy>=1.14.2",
-        "torch==1.7.1",
-        "torchvision==0.8.2",
+        "torch==1.8.1",
+        "torchvision==0.9.1",
         "tensorboardX>=1.7",
         "tqdm>=4.19.9",
         "matplotlib>=2.1.0",
