Вы можете использовать класс tf.train.Saver
, чтобы сохранить модель в виде файла ckpt и использовать его позже:
saver = tf.train.Saver()
with tf.Session() as sess:
#your code here
save_path = saver.save(sess, "/model_path/model.ckpt")
Для восстановления переменных:
saver.restore(sess, "/your_path/model.ckpt")
Полная документация: https://www.tensorflow.org/guide/saved_model