added version in __init__.py

2018-12-11 12:12:08 +01:00
parent d3fcec1a3e
commit ce52177638
3 changed files with 4 additions and 4 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,5 @@
-# This installs Pytorch for CUDA 8 only. If you are using a newer version,
-# please visit http://pytorch.org/ and install the relevant version.
-torch>=0.4.1,<0.5.0
+# PyTorch
+torch>=0.4.1
 # progress bars in model download and training scripts
 tqdm
 # Accessing files from S3 directly.