keras(tensorflow后端)给模型建立单独的graph

因为要在一个项目中同时导入多个模型,需要给keras模型新建个graph,但是按照tensorflow新建graph的方式一直类似的报错:

ValueError: Tensor Tensor("Sigmoid_2:0", shape=(?, 17), dtype=float32) is not an element of this graph.

原因大概是多线程、分布式环境下,恢复Model时的Tensor Graph和生成Model时不同
百度了下解决方法,完美解决:

定义模型结构,得到模型basemodel
g3=tf.get_default_graph()
with g3.as_default():
    basemodel.load_weights(modelPath)

你可能感兴趣的:(keras(tensorflow后端)给模型建立单独的graph)