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