В Tensorflow как заморозить сохраненную модель

Наверное, это очень простой вопрос...

Но как преобразовать файлы контрольных точек в один файл .pb. Моя цель - обслуживать модель, используя, вероятно, С++

Это файлы, которые я пытаюсь преобразовать.

Контрольные точки

В качестве примечания я использую tflearn с tensorflow.

Изменить 1: я нашел статью, в которой объясняется, как это сделать: https://blog.metaflow.fr/tensorflow-how-to-freeze-a-model-and-служить-это-с-а-python-api-d4f3596b3adc

Проблема в том, что я застрял со следующей ошибкой

KeyError: "The name 'Adam' refers to an Operation not in the graph."

Как это исправить?

Изменить 2: возможно, это прольет свет на проблему.

Слой регрессии

Ошибка, которую я получаю, исходит от уровня регрессии, если я использую: sgd. Я получу

KeyError: "The name 'SGD' refers to an Operation not in the graph."

python tensorflow tensorflow-serving tflearn
person redb    schedule 09.08.2017    source источник
comment
вы решаете то же самое, ссылаясь. github.com/tflearn/tflearn/issues/605   -  person Ishant Mrinal    schedule 09.08.2017

Ответы (1)


arrow_upward
3
arrow_downward

Учебник по https://blog.metaflow.fr/tensorflow-how-to-freeze-a-model-and-serve-it-with-a-python-api-d4f3596b3adc работает просто отлично .

Проблема заключалась в том, что я загружал модель с помощью tensorflow вместо tflearn.

Итак... вместо:

tf.train.import_meta_graph(...)

We do:

model.load(...)

TFLearn умеет правильно анализировать граф.

person redb    schedule 09.10.2017
comment
model.load(...)? можно подробнее? - person Morse; 04.05.2018
comment
model.load(...) исходит от TFLearn, а не от Tensorflow. Я смешивал два API. - person redb; 05.05.2018
comment
Проверьте здесь: tflearn.org/#quick-overview, в первом примере кода они создают модель делаю model = tflearn.DNN(net). Из этого вы можете сделать model.load - person redb; 06.05.2018
comment
Этот просит построить сеть с нуля, я хотел сохранить модель, построенную программой, а позже иметь разностную программу только для вызова модели из прогнозов. Не получается просто импортировать эти файлы, сгенерированные tflearn (мета, индекс...) - person Morse; 07.05.2018