由于图形问题，无法在Keras fit_generator中运行Universal Sentence Encoder

Question

我在这样的生成器中使用USE。

embed = hub.Module("https://tfhub.dev/google/universal-sentence-encoder-large/3")

sess = tf.Session()
K.set_session(sess)

sess.run(tf.global_variables_initializer())
sess.run(tf.tables_initializer())

def generator():
    while True:
        batch_features = sess.run(embed(...))
        batch_labels = ...
        yield batch_features, batch_labels

在定义Keras模型并调用fit_generator之后，

model.fit_generator(generator(), steps_per_epoch=..., epochs=...)

有错误

RuntimeError: Module must be applied in the graph it was instantiated for.

似乎图形在generator（）中已更改。我尝试了几种关于图形的操作，但没有成功。请帮我解决这个问题。