本文記錄 keras 使用過程中遇到的一些問題,不定期更新。
1、查看 keras 版本
python -c "import keras; print keras.__version__"
2、更新 keras
sudo pip install --upgrade keras
3、修改 keras 的 backend
方法1
在users
文件夾下新建.keras
文件夾,在文件夾下新建keras.json
文件,json內容爲:
{
"image_dim_ordering": "tf",
"epsilon": 1e-07,
"floatx": "float32",
"backend": "tensorflow"
}
- 這其中的backend可以是
tensorflow
或者theano
,當然前提是你已經安裝好了。
但是上面這種方式有一個弊端就是每次當你想切換backend的時候都需要手動修改這個文件,所以有一種能夠動態修改的方式不是更好嗎?請看下面的方法:
方法2
import os
os.environ['KERAS_BACKEND'] = 'tensorflow'
import keras.backend as K
K.set_image_dim_ordering('tf')
- 首先載入os庫,將keras的backend修改爲tensorflow,再將keras的backend加載,設置order爲tensorflow格式。
方法3
KERAS_BACKEND=theano python -c "from keras import backend"