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