add tensorflow and keras version

cl0ver012 · Jan 9, 2019 · fcb584b · fcb584b
1 parent 90956f2
commit fcb584b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -3,8 +3,8 @@ This repository is to reproduce Deep Image Matting.
 
 ## Dependencies
 - [NumPy](http://docs.scipy.org/doc/numpy-1.10.1/user/install.html)
-- [Tensorflow](https://www.tensorflow.org/versions/r0.8/get_started/os_setup.html)
-- [Keras](https://keras.io/#installation)
+- [Tensorflow 1.9.0](https://www.tensorflow.org/)
+- [Keras 2.1.6](https://keras.io/#installation)
 - [OpenCV](https://opencv-python-tutroals.readthedocs.io/en/latest/)
 
 ## Dataset

diff --git a/train.py b/train.py
@@ -76,5 +76,5 @@ def on_epoch_end(self, epoch, logs=None):
                         verbose=1,
                         callbacks=callbacks,
                         use_multiprocessing=True,
-                        workers=int(get_available_cpus() / 2)
+                        workers=2
                         )