Как преобразовать контрольную точку в модель .pb для развертывания модели? - PullRequest
Новая
       19

Как преобразовать контрольную точку в модель .pb для развертывания модели?

1 голос
/ 09 апреля 2019

Я обучил модель перевода языка seq2seq по тензорному потоку и сохранил в виде контрольных точек следующие файлы в папке моего поезда.

  • translate.ckpt-157450.data-00000-of-00001
  • translate.ckpt-157450.index
  • translate.ckpt-157450.meta и
  • файл контрольных точек

Теперь я хочупреобразовать его в файл protobuf (.pb) для целей развертывания.Вот некоторый код, который я использую: 

import tensorflow as tf
meta_path = "/home/i9/L-T_Model_Training/01_Apr_model/train/translate.ckpt-157450.meta"
with tf.Session() as sess:
saver = tf.train.import_meta_graph(meta_path)
saver.restore(sess, tf.train.latest_checkpoint('.'))
output_node_names =[n.name for n in tf.get_default_graph().as_graph_def().node]
frozen_graph = tf.graph_util.convert_variables_to_constants(sess, sess_graph_def, output_node_names)
with open("output_graph.pb", "wb") as f:
    f.write(frozen_graph.SerializeToString())

Я запускаю этот код в своей папке поезда.Он показывает мне ошибку: ValueError: Невозможно загрузить save_path, когда он отсутствует.

Я также пробовал скрипт freeze_graph.py, но не смог получить модель.

...