TensorFlow入门(九)使用 tf.train.Saver()保存模型

关于模型保存的一点心得 saver = tf.train.Saver(max_to_keep=3) 在定义 saver 的时候通常会定义最多保存模型的数量,通常来讲,若是模型自己很大,咱们须要考虑到硬盘大小。若是你须要在当前训练好的模型的基础上进行 fine-tune,那么尽量多的保存模型,后继 fine-tune 不必定从最好的 ckpt 进行,由于有可能一会儿就过拟合了。可是若是保存太多,硬盘
相关文章
相关标签/搜索