TensorFlow01:保存模型

tf.train.Saver(var_list=None,max_to_keep=5)
     var_list:指定将要保存和还原的变量,可以是一个列表或字典
     max_to_keep:指示要保留的最近检查点文件的最大数量,创建新的文件时删除较旧的文件
saver.save(sess,”xxx.ckpt”)
saver.restore(sess,”xxx.ckpt”)

猜你喜欢

转载自www.cnblogs.com/jumpkin1122/p/11521027.html