Merge branch 'master' of github.com:jhfjhfj1/autokeras

keras-team · Feb 5, 2019 · 342db0b · 342db0b
2 parents 6189579 + 00b21c3
commit 342db0b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
     name='autokeras',
     packages=find_packages(exclude=('tests',)),
     install_requires=['scipy==1.2.0',
-                      'torch==0.4.1',
+                      'torch==1.0.1',
                       'torchvision==0.2.1',
                       'numpy==1.15.4',
                       'keras==2.2.4',
@@ -15,7 +15,7 @@
                       'tensorflow==1.12.0',
                       'imageio==2.4.1',
                       'requests==2.21.0',
-                      'lightgbm==2.2.2',
+                      'lightgbm==2.2.3',
                       'pandas==0.23.4',
                       'librosa==0.6.2',
                       'numba',