方法一:
vim ~/.keras/keras.json
修改 -backend:tensorflow/theano
若没有该文件,则添加如下内容:
"floatx": "float32",
"epsilon": 1e-07,
"backend": "TensorFlow",
"image_data_format": "channels_last"
方法二:
import os
os.environ['KERAS_BACKEND'] = 'tensorflow'
import keras.backend as K
K.set_image_dim_ordering('tf')
博客介绍了修改Keras后端的两种方法。方法一是通过vim编辑~/.keras/keras.json文件,修改backend为tensorflow或theano;若文件不存在,则需添加相应内容。
2566

被折叠的 条评论
为什么被折叠?



