TensorFlow | tf.train.Saver( )

TensorFlow ——tf.train.Saver( )用于模型的保存

1、先实例化saver

      saver=tf.train.Saver( ) 创建一个Saver( )对象

      saver=tf.train.Saver( max_to_keep=5) 在这里,创建对象可以选择保存近几次的模型,比如这里是只保存近5次的模型

2、保存模型

      saver.save(sess,r"路径",global_step=step)

发布了26 篇原创文章 · 获赞 9 · 访问量 8224

猜你喜欢

转载自blog.csdn.net/weixin_41664064/article/details/89395882