Как мне сделать модель предварительного обучения в Tensorflow?

Question

Следующие файлы были созданы с помощью tf-slim

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

Есть ли способ интегрировать этот файл в один .ckpt файл, предоставленный tenorflow

Answer 1

Предварительно обученные модели, предоставляемые Tensorflow, представляют собой не один файл .ckpt, а обычно два (.ckpt.data, .ckpt.index) или три (.ckpt.data, .ckpt.meta, .ckpt.index).Вы можете загрузить переменную просто, попробовав это:

with tf.Session() as sess:
    saver = tf.train.Saver()
    saver.restore(sess, "/tmp/model.ckpt")

не потому, что существует файл с именем model.ckpt, а потому, что из saver.py он автоматически находит файлы для восстановления переменных.Чтобы восстановить график, вы можете использовать это:

tf.train.import_meta_graph('/tmp/model.ckpt.meta')

Вы можете сослаться на TensorFlow, почему после сохранения модели есть 3 файла? для получения дополнительной информации.