keras第二课:后端函数

一、修改backend的方法

(一)找到.keras文件夹,在文件夹下新建keras.json文件,json内容为:

{
    "image_dim_ordering": "tf", 
    "epsilon": 1e-07, 
    "floatx": "float32", 
    "backend": "tensorflow"
}

这种方式有一个弊端就是每次当你想切换backend的时候都需要手动修改这个文件

(二)动态修改

import os
os.environ['KERAS_BACKEND'] = 'tensorflow'
import keras.backend as K
K.set_image_dim_ordering('tf')

二、backend函数

见https://www.cnblogs.com/aibbtcom/p/8541794.html

posted on 2018-09-25 21:13  4c4853  阅读(139)  评论(0编辑  收藏  举报

导航