Sej*_*air 4 python restore tensorflow
我有一个模型,我每10次迭代就会保存一次.所以,我在保存的目录中有以下文件.
checkpoint model-50.data-00000-of-00001 model-50.index model-50.meta
model-60.data-00000-of-00001 model-60.index model-60.meta
等等达到100.我只需加载模型-50.因为我在70次迭代后得到了NaN值.通过deafault,当我恢复保护程序时,将寻找最终的检查点.那么,我怎么能专门加载模型-50.请帮助,否则,我必须从头开始运行模型增益,这很费时间.
由于您使用tf.train.Saver的是函数restore(),因此您可以使用这些last_checkpoints函数来获取所有可用检查点的列表.你也可以看到model-50,并model-60在此列表中.
选择正确的模型,并将其直接传递给restore()这样,
saver.restore(sess, ckpt_path)
| 归档时间： |
|
| 查看次数： |
3742 次 |
| 最近记录： |