Как сохранить модель декодера Tensorflow Encoder?

Question

Я следовал этому учебнику о построении модели перевода языка кодировщик-декодер и построил ее для моего родного языка.

Теперь я хочу сохранить ее, развернуть на облачном движке ML и сделатьпредсказания с HTTP-запросом.

Я не смог найти четкий пример того, как сохранить эту модель,

Я новичок в ML и нашел Руководство по сохранению TF v сбивает с толку ..

Есть ли способ сохранить эту модель, используя что-то вроде tf.keras.models.save_model

Answer 1

Вы можете сохранить модель Keras в формате Keras в формате HDF5, см .:

https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model

Вы захотите сделать что-то вроде:

import tf.keras
model = tf.keras.Model(blah blah)
model.save('my_model.h5')

Если выперейти на TF 2.0, проще построить модель в tf.keras и развернуть ее с использованием формата TF SavedModel.Этот учебник 2.0 показывает использование предварительно обученной модели tf.keras, сохранение модели в формате SavedModel, развертывание в облаке и затем выполнение HTTP-запроса для прогноза:

https://www.tensorflow.org/beta/guide/saved_model

Answer 2

Создайте заставку поезда после открытия сеанса и после окончания тренировки сохраните модель:

with tf.Session() as sess:
     saver = tf.train.Saver()
     # Training of the model
     save_path = saver.save(sess, "logs/encoder_decoder")
     print(f"Model saved in path {save_path}")