Не удается восстановить модель тензорного потока с помощью Saver - PullRequest
0 голосов
/ 25 апреля 2020

Я следую этому руководству по использованию класса Saver в Tensorflow версии 1.

Сначала я сохраняю модель:

with tf.Session() as sess:
  init = tf.global_variables_initializer()
  sess.run(init)
  saver.save(sess, "./saved_model/tf/model", global_step=0)

, которая дает мне эти файлы:

$ ls saved_model/tf
checkpoint  model-0.data-00000-of-00001  model-0.index  model-0.meta

Но когда я пытаюсь восстановить сеанс, я получаю сообщение об ошибке:

with tf.Session() as sess:
  saver.restore(sess, "./saved_model/tf/model")

---------------------------------------------------------------------------

ValueError                                Traceback (most recent call last)

<ipython-input-37-01cbbefb52af> in <module>()
      1 with tf.Session() as sess:
----> 2   saver.restore(sess, "./saved_model/tf/model")

/tensorflow-1.15.2/python3.6/tensorflow_core/python/training/saver.py in restore(self, sess, save_path)
   1280     if not checkpoint_management.checkpoint_exists_internal(checkpoint_prefix):
   1281       raise ValueError("The passed save_path is not a valid checkpoint: " +
-> 1282                        checkpoint_prefix)
   1283 
   1284     logging.info("Restoring parameters from %s", checkpoint_prefix)

ValueError: The passed save_path is not a valid checkpoint: ./saved_model/tf/model

Что я делаю не так? К сожалению, документация TF по этой функции мало помогает.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...