ubuntu下切换keras的后端backend

2017-03-05  本文已影响0人  MarkSJL

1、keras后端配置是在文件keras.json 这个文件中进行的,因此找到目录,不存的话会自动创建

sudo gedit ~/.keras/keras.json

2、theano作为backend

{

"image_dim_ordering":"th",

"epsilon":1e-07,

"floatx":"float32",

"backend":"theano"

}

3、TensorFlow 作为backend

{

"image_dim_ordering":"tf",

"epsilon":1e-07,

"floatx":"float32",

"backend":"tensorflow"

}

ref: Keras 切换后端(Theano和TensorFlow)

上一篇下一篇

猜你喜欢

热点阅读