代码之家 › 专栏 › 技术社区 › tigertang

如何将tensorflow保存的_模型转换为ckpt?

converters model tensorflow

0

tigertang · 技术社区 · 6 年前

我只有接受 ckpt 文件夹中提供了预先培训的模型 saved_model 格式。有什么方法可以转换吗 保存的_模型 ckpt公司

0 回复 | 直到 6 年前

1

0

tigertang 6 年前

我想明白了:

在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")