问题:
Flask部署keras模型服务时,出现以下错误:
ValueError: Tensor Tensor("dense_1/Sigmoid:0", shape=(?, ?, 1), dtype=float32) is not an element of this graph.
解决方案:
global sess
global graph
sess = tf.Session()
graph = tf.get_default_graph()
set_session(sess)
#加载模型
relation_extract.load_model()
#下面是接口函数中调用
with graph.as_default():
set_session(sess)
result= relation_extract.predict(text)
网友评论