Keras默認使用的後端模式(Theano/TensorFlow)設置

其實就是在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

最後保存就可以了

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