TensorFlow, обучение производит 3 файла ckpt, какой из них правильный?(Music_RNN_RBM) - PullRequest
0 голосов
/ 09 марта 2019

Я даю выстрел в Music_RNN_RBM

Я обучил модель и, как сказано в инструкции, создается файл ckpt, проблема в том, что в каталоге для каждой сохраненной эпохи у меня есть три .ckpt .data .index и .meta Пример:

model.ckpt-151800.data-00000-of-00001
model.ckpt-151800.index
model.ckpt-151800.meta

Теперь следующим шагом будет создание корыта

python rnn_rbm_generate.py <path_to_ckpt_file>

Я пытался сделать это для каждого типа файла

python3 rnn_rbm_generate.py model.ckpt-151800.index

Но я получаю эту ошибку каждый раз

Traceback (most recent call last):
  File "rnn_rbm_generate.py", line 52, in <module>
    main(sys.argv[1])
  File "rnn_rbm_generate.py", line 43, in main
    saver.restore(sess, ckpt.model_checkpoint_path)
AttributeError: 'NoneType' object has no attribute 'model_checkpoint_path' 

Я не понимаю, что должен делать

Ответы [ 2 ]

0 голосов
/ 09 марта 2019

Есть функция для загрузки последней .Используйте его для загрузки и создания надлежащей сохраненной модели, контрольные точки используются в качестве резервной копии или для продолжения обучения после, а не для обслуживания.Упомянутый prouast файла 'checkpoints' загружается упомянутой мной функцией, это информация о файлах, которые создаются и не могут быть загружены.Вопросы задавайте

РЕДАКТИРОВАТЬ: кроме того код кода в вашем репо указывает, что вы также можете предоставить полную папку, а не файл.

0 голосов
/ 09 марта 2019

Как вы обнаружили, каждая контрольная точка состоит из .data, .meta и index файла.

Однако TensorFlow также создает файл checkpoints, который отслеживает всеконтрольные точки в выходном каталоге.Он содержит ссылку model_checkpoint_path на самую последнюю контрольную точку, которую ищет ваша модель.

Поэтому, чтобы ответить на ваш вопрос, указанный вами путь должен содержать этот checkpoints файл.

...