Как получить промежуточные значения из tf.saved_model в tenorflow2? - PullRequest
Новая
       40

Как получить промежуточные значения из tf.saved_model в tenorflow2?

0 голосов
/ 22 апреля 2020

Я получил tf.saved_model, который предсказывает выходные данные по желанию, но мне нужны логиты из модели для некоторых вычислений. Как я могу получить промежуточные значения из графика tf.saved_model? Ниже моя тестовая реализация, которая не работает. 

import tensorflow as tf

MODEL_DIR = '/path/to/model_dir'

model = tf.saved_model.load(MODEL_DIR)
model = model.signatures.get('serving_default')

graph = model.graph

i = graph.inputs[0]
o = graph.get_operations()[-2].outputs[0]


with tf.compat.v1.Session(graph=graph) as sess:
  y_out = sess.run(o, feed_dict={i: [[1,2,3,4]]})
  print(y_out)

Для приведенного выше кода я получаю следующую ошибку. 

TypeError: Tensor is unhashable if Tensor equality is enabled. Instead, use tensor.experimental_ref() as the key.

Как правильно это сделать?

...