当使用带有TF后端的Keras时,我可以使用K.get_session()
创建一个新的TF会话,如下所示:
from keras import backend as K
tf_session = K.get_session()
Keras后端是否有类似的功能来启动交互式TF会话?在文档中找不到任何内容。即使K.get_session()
没有记录,所以我想也许我在某处错过了它。
答案 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模型创建自定义的会话和图形。