Keras默认使用的后端模式(Theano/TensorFlow)设置

来源:互联网 发布:小型公司网络布线 编辑:程序博客网 时间:2024/05/04 04:54

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

用theano的话,keras.json写入

{    "image_dim_ordering": "th",     "epsilon": 1e-07,     "floatx": "float32",     "backend": "theano"}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

用thesorflow的话,keras.json写入

{    "image_dim_ordering": "tf",     "epsilon": 1e-07,     "floatx": "float32",     "backend": "tensorflow"}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

最后保存就可以了

原创粉丝点击