Merge pull request #13 from valohai/pin-keras

Pin keras version
valohai · Feb 13, 2021 · 82ebe12 · 82ebe12
2 parents 552bff5 + 7aef932
commit 82ebe12
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements-cpu.txt b/requirements-cpu.txt
@@ -1,7 +1,7 @@
 appdirs==1.4.3
 h5py>=2.6.0
 imageio==2.5.0
-Keras>=2.3.1
+Keras==2.3.1
 numpy>=1.12.0
 packaging==16.8
 pyparsing==2.2.0

diff --git a/requirements-gpu.txt b/requirements-gpu.txt
@@ -1,7 +1,7 @@
 appdirs==1.4.3
 h5py>=2.6.0
 imageio==2.5.0
-Keras>=2.3.1
+Keras==2.3.1
 numpy>=1.12.0
 packaging==16.8
 pyparsing==2.2.0