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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u010122972/article/details/79093634

因为要在一个项目中同时导入多个模型,需要给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)

猜你喜欢

转载自blog.csdn.net/u010122972/article/details/79093634
今日推荐