keras 随笔记

本文记录 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"

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章