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