07 2018 档案

TensorFlow保存和载入模型
摘要：首先定义一个tf.train.Saver类： 其中，max_to_keep参数设定只保存最后一个参数，默认值是5，即保存最后5个模型，如果设置成0，训练过程中的所有模型都会被保存。 模型训练好以后，保存模型： 其中，sess是Session，ckpt_dir + "/nn_model.ckpt"是保 阅读全文
posted @ 2018-07-31 14:17 MSTK 阅读(429) 评论(0) 推荐(0)