keras-backend修改配置

来源:互联网 发布:seo站长工具下载 编辑:程序博客网 时间:2024/05/16 07:21

方法1:修改keras的配置文件,~/.keras/keras.json

{
    "backend": "tensorflow",
    "epsilon": 1e-07,
    "floatx": "float32",
    "image_data_format": "channels_last"
}

可以直接将backend项修改为tensorflow或theano,如果直接修改有问题,可以在文本中修改再整体拷贝修改该配置文件即可。

方法2:通过命令修改

KERAS_BACKEND = tensorflow python -c "from keras import backend"

这种修改方式只是临时修改了backend,可以在程序中加入

import os

os.environ['KERAS_BACKEND']= 'theano'

原创粉丝点击