python - Keras load_model 和 get_default_graph.finalize()

Question

我在尝试完成加载的图表时遇到了麻烦。我的管道是下一个：

import tensorflow as tf
from keras.models import load_model

model = load_model("model")
model.load_weights("weights")
graph = tf.get_default_graph()
graph.finalize()

但我有这样的错误：

RuntimeError: Graph is finalized and cannot be modified.

问题是为什么会发生？我正在尝试在加载模型和权重后完成图形，它应该可以正常工作，所以我无法获得这种行为。

标签： pythontensorflowkeras

Answer 1

如果您尝试在多线程中加载模型，则需要清除会话

from keras import backend as K
K.clear_session()