Tensorflow: хранение в моделях - PullRequest
0 голосов
/ 27 мая 2018

Я новичок в tf и хочу сохранить активацию в модели.

Я пытаюсь построить классификатор изображений и прогнозирую одно изображение, используя sess.run(activation,feed_dict={X:npArray}).

Как сохранить активацию и X в модели и восстановить их обратно?

Я использую что-то подобное для восстановления

y_pred = tf.get_default_graph().get_tensor_by_name('y_pred:0')
X = tf.get_default_graph().get_tensor_by_name('X:0')

и затем использования

output_label = sess.run(y_pred, feed_dict={X: npArray}) 

для прогнозирования метки класса, но независимо от того, что является вводом npArray, output_label не изменяется.Если даже сохраненная модель изменяется, output_label не изменяется.Это может означать, что переменная y_pred и заполнитель X не являются

  1. или правильно хранятся в модели;или
  2. они не восстановлены должным образом

Любая помощь будет оценена.Спасибо

...