会话的前面加:saver=tf.train.Saver()
会话里面的最下方加: saver.save(sess,‘ckpt/XXHG.ckpt’)
模型会直接保存在和脚本同一路径下,
注意:模型的格式是ckpt
【TensorFlow】模型保存
最新推荐文章于 2025-05-19 20:55:27 发布
会话的前面加:saver=tf.train.Saver()
会话里面的最下方加: saver.save(sess,‘ckpt/XXHG.ckpt’)
模型会直接保存在和脚本同一路径下,
注意:模型的格式是ckpt