У меня есть код, как показано ниже. Работает как задумано. Но я хочу сохранить значение переменной embeddings
в памяти. После выполнения кода я не вижу переменную embeddings
в моем variable explorer
. Я бы хотел избежать записи на диск. Есть ли простой способ сделать это?
import tensorflow_hub as hub
import tensorflow as tf
elmo = hub.Module("https://tfhub.dev/google/elmo/2", trainable=True)
x = ["she is the love of my life"]
# Extract ELMo features
embeddings = elmo(x, signature="default", as_dict=True)["elmo"]
embeddings.shape
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
sess.run(tf.tables_initializer())
print (sess.run(embeddings))