saver.save如何使用?
时间: 2023-11-07 17:24:20 浏览: 88
TensorFlow 中 训练模型 train-saver.zip
saver.save是 TensorFlow 模型保存的方法,用于将训练好的模型保存到磁盘上。
具体使用方法如下:
1. 创建一个 tf.train.Saver 对象:saver = tf.train.Saver()
2. 在训练完成后,通过 saver.save() 方法保存模型:
saver.save(sess, save_path)
其中,sess 是当前 TensorFlow 会话对象, save_path 是模型保存的路径。
阅读全文