как сохранить контрольную точку Tensorflow Keras в файл pb - PullRequest
0 голосов
/ 26 сентября 2018

Кто-нибудь знает, как сохранить файл контрольных точек tenorflow keras в двоичный файл .pb для обслуживания?

Я не уверен на 100%, как это реализовать.

Нужно ли объявлять tf.keras.backend.get_session() для сохранения графика и файла pb?

В сумме 1. Способ сохранения файла pb с помощью tf.keras или 2. Любые подходы с использованием tf.keras.estimator.model_to_estimator(keras_model=model)

Спасибо

1 Ответ

0 голосов
/ 26 сентября 2018

вот ссылка на то, как использовать керас с tenorflow . В основном, вы должны получить модель тензорного потока из keras (K.get_session () и K.get_graph (). Graph) и использовать Тензор Поток Метод чтобы сохранить его.

...