tensorflow中使用save和restore保存和恢复模型
我们在训练模型过程中,有时训练一段时间后,往往想要在验证集上验证一下,模型是否存在过拟合,然后视验证情况,再选择继续训练还是修改模型参数。这时tensorflow提供的Saver类,就能很好的帮助到我们。 当我们保存一个模型到指定路径后,还目录下将会出现四种类型的文件: checkpoint: 具有最近检查点列表的协议缓存区 .data: 保存模型中的变量 .index: 标识检查点 ....
原创
2018-06-06 15:10:02 ·
9884 阅读 ·
0 评论