update requirements.txt

mengxiangming · Jul 22, 2020 · 428a12d · 428a12d
1 parent e1f89de
commit 428a12d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,12 +1,11 @@
 tqdm==4.26.0
 numpy==1.14.3
-torchvision==0.2.1
+torch>=1.5
+torchvision>=0.4
 gpytorch==0.1.0rc4
 tabulate==0.8.2
 scipy==1.1.0
 setuptools==39.1.0
 matplotlib==2.2.2
-torch==1.0.0
 Pillow>=6.2.2
-datasets==0.0.9
 scikit_learn==0.20.2