Как преобразовать тензор потока сохраненных_моделей в ckpt?

Question

У меня есть только API, который принимает папку ckpt, но предварительно обученная модель предоставляется в формате saved_model. Есть ли способы конвертировать saved_model в ckpt?

Answer 1

Я понял это:

Под TensorFlow 1.15:

import tensorflow as tf

with tf.Session() as sess:
    imported = tf.saved_model.load(
        sess, ["serve"], "${saved_model_path}")
    saver = tf.train.Saver(max_to_keep=1)
    saver.save(sess, "${model_name}.ckpt")