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)
1、先实例化saver
saver=tf.train.Saver( ) 创建一个Saver( )对象
saver=tf.train.Saver( max_to_keep=5) 在这里,创建对象可以选择保存近几次的模型,比如这里是只保存近5次的模型
2、保存模型
saver.save(sess,r"路径",global_step=step)