from tensorflow.python.framework.graph_util import convert_variables_to_constants
saver = tf.train.Saver(tf.trainable_variables())
saver.restore(sess
tensorflow 将ckpt保存为pb文件
最新推荐文章于 2023-01-30 17:24:22 发布
这篇博客介绍了如何在TensorFlow 1.14或1.15版本中,将模型的ckpt权重文件转换为便于部署的.pb格式。详细步骤包括了利用tf.train.export_meta_graph和tf.train.Saver等工具进行模型的保存和转换。
摘要由CSDN通过智能技术生成