1. 程式人生 > >keras(tensorflow後端)給模型建立單獨的graph

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)