Keras后端:启动交互式TF会话

时间:2017-11-30 15:25:34

标签: python tensorflow keras

当使用带有TF后端的Keras时,我可以使用K.get_session()创建一个新的TF会话,如下所示:

from keras import backend as K
tf_session = K.get_session()

Keras后端是否有类似的功能来启动交互式TF会话?在文档中找不到任何内容。即使K.get_session()没有记录,所以我想也许我在某处错过了它。

1 个答案:

答案 0 :(得分:0)

您可以通过执行此操作来定义会话

graph = tf.Graph() # Tensorflow graph
with graph.as_default():
    session = tf.Session() # Tensorflow session
    with session.as_default():
        model.load('model.h5')
        # or define your model or use predict or eval methods... 

像这样,您可以为Keras模型创建自定义的会话和图形。