码迷,mamicode.com
首页 > 其他好文 > 详细

Keras 切换后端(Theano和TensorFlow)

时间:2017-05-20 00:02:00      阅读:245      评论:0      收藏:0      [点我收藏+]

标签:blank   name   test   描述   code   ref   use   backend   att   

中文文档的描述:keras中文文档,切换后端

其实就是在C:\Users\75538(75538是我的windos用户名字,找你对应的用户名就行)下有个文件夹.keras,里面有keras.json文件,改一下里面的内容就好了,如果没有文件夹和文件,手动创建就行。

用theano的话,keras.json写入

{
    "image_dim_ordering": "th", 
    "epsilon": 1e-07, 
    "floatx": "float32", 
    "backend": "theano"
}

用thesorflow的话,keras.json写入

{
    "image_dim_ordering": "tf", 
    "epsilon": 1e-07, 
    "floatx": "float32", 
    "backend": "tensorflow"
}

最后保存就可以了

Keras 切换后端(Theano和TensorFlow)

标签:blank   name   test   描述   code   ref   use   backend   att   

原文地址:http://www.cnblogs.com/huang-hao/p/6880547.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!