Tensorflow:保存,加载模型

有时常常需要将训练的模型保存下来,用以预测或之后接着训练。 1. 保存模型 主要是通过红色框圈出的两句话来保存模型的,下面是对这两句话的解释: saver = tf.train.Saver() 在定义saver类时,可以指定需要保存的变量,若不指定,则是保存计算图中所有的变量,即像我这样处理。max_to_keeo参数是指最多保存模型的个数,此处设置为5,即最多保存5个模型。 saver.save
相关文章
相关标签/搜索