如何将 tensorflow saved_model 转换为 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")