会话的前面加:saver=tf.train.Saver()
会话里面的最下方加: saver.save(sess,‘ckpt/XXHG.ckpt’)
模型会直接保存在和脚本同一路径下,
注意:模型的格式是ckpt
【TensorFlow】模型保存
猜你喜欢
转载自blog.csdn.net/ZHANGHUIHUIA/article/details/84554073
今日推荐
周排行