как сохранить контрольную точку Tensorflow Keras в файл pb

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

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

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

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

Спасибо


keras tensorflow tensorflow-datasets tensorflow-estimator
person RSBS    schedule 26.09.2018    source источник

Ответы (1)


arrow_upward
1
arrow_downward

вот ссылка на то, как использовать keras с tensorflow в основном вы должны получить модель тензорного потока от keras (K.get_session() и K.get_graph().graph) и использовать метод tensorflow, чтобы сохранить его.

person willian ver valem    schedule 26.09.2018
comment
Я уже видел эту статью, но когда я сохранил модель keras, ее нужно снова перезагрузить через модель keras, но я не уверен, как это сделать. Следующие инструкции бесполезны в отношении файлов сохранения pb. Есть ли здесь какой-нибудь простой пример сохранения pb tf.keras? - person RSBS; 27.09.2018