深度学习框架keras:如何设置后端Tensorflow/theano

别来无恙 提交于 2019-11-29 23:45:41

修改keras配置文件,可以更改keras后端依赖的框架:

sudo gedit ~/.keras/keras.json
{
    "epsilon": 1e-07, 
    "floatx": "float32", 
    "image_data_format":"channels_last" ,
    "backend": "tensorflow",
    "image_dim_ordering":"th"
}

把tensorflow改为theano,
同时把channels_last改为channels_first,
即可切换后端框架,如下:

{
    "image_data_format": "channels_first",
    "epsilon": 1e-07,
    "backend": "theano",
    "floatx": "float32"
    "image_dim_ordering":"th"
}
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!