Экспорт модели seq2seq для обслуживания Tensorflow с именами входных или выходных тензоров - PullRequest
0 голосов
/ 14 февраля 2019

Я использую официальную библиотеку seq2seq для генерации моделей.Сгенерированные файлы моделей: (* ​​1003 *, .meta, .index, .pbtxt, checkpoint)

Поскольку файлы не в формате SavedModel, я не смог использовать его дляTensorflow Serving.

После изучения я получаю две идеи:

  1. Использование созданных файлов для создания файла frozen(.pb).Но я не смог найти --output_node_names, как ожидалось freeze_graph.py

  2. Добавить estimator.export_savedmodel внутри train.py, чтобы автоматически создать график в ожидаемом формате, используя это .Но я не смог найти names of input tensors, как ожидалось от serving_input_receiver_fn()

Для данного кода в master ветке, я хотел бы знатьимена input и output тензоров.С этим я мог сделать любой из двух вышеуказанных методов.

Как я мог их найти?Кроме того, какие файлы или части кода необходимо изменить в любом случае?

...