Update BindsNET requirements to the new PyTorch version

Hananel-Hazan · Hananel-Hazan · commit 2cfaf8a3c378 · 2020-07-10T19:27:13.000-04:00
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -3,4 +3,4 @@ repos:
     rev: master
     hooks:
     - id: black
-      language_version: python3.7
+      language_version: python3.8
diff --git a/requirements.txt b/requirements.txt
@@ -1,8 +1,8 @@
 foolbox
 scipy>=1.1.0
 numpy>=1.14.2
-torch==1.4.0
-torchvision==0.5.0
+torch>=1.5.1
+torchvision>=0.6.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.4.0",
-        "torchvision==0.5.0",
+        "torch>=1.5.1",
+        "torchvision>=0.6.1",
         "tensorboardX>=1.7",
         "tqdm>=4.19.9",
         "matplotlib>=2.1.0",
