在代码导包位置加两句代码
import os
os.environ[‘KERAS_BACKEND’] = ‘tensorflow’
没有改 keras.json文件(当前是’th’, theano)也可以
终于不爆内存了。。。。。。。。。。。。。。。。happy again--------
import os
os.environ[‘KERAS_BACKEND’] = ‘theano’
keras.json 文件
{
“image_dim_ordering”: “th”,
“epsilon”: 1e-07,
“floatx”: “float32”,
“backend”: “theano”
}
{
“image_dim_ordering”: “tf”,
“epsilon”: 1e-07,
“floatx”: “float32”,
“backend”: “tensorflow”
}