Очень простой пример для воспроизведения:
import tensorflow as tf
import tensorflow_hub as hub
import tensorflow_text
module = hub.load("https://tfhub.dev/google/universal-sentence-encoder-multilingual-qa/3")
for i in range(50):
response_embeddings = module.signatures['response_encoder'](
input=tf.constant(["I am happy", "I am not happy"]),
context=tf.constant(["I am happy", "I am not happy"]))
t = response_embeddings["outputs"].numpy()
print(t.shape)
После запуска этого последнего l oop примерно 3 раза, он просто застревает навсегда. Это известная проблема ? Я использую ноутбук Jupyter. Я использую TF 2.3 на ЦП в MacOS.