如何加载非最新的 TensorFlow 检查点？答案

【问题标题】：How do I load a non-latest Tensorflow checkpoint?如何加载非最新的 TensorFlow 检查点？
【发布时间】：2022-11-03 11:11:32
【问题描述】：

我每训练 1000 步就创建一个检查点，我的检查点目录中有 16 个文件。然而，似乎当我想检索最新的它时，它正在恢复到它的预训练状态。我假设与未记录后续检查点存在的摘要日志有关。

chkpt.restore(tf.train.latest_checkpoint(chkpt_dir))
# fit(train_ds, test_ds, steps=100000)

for i in range(10):
    ex_input, ex_output = next(iter(test_ds.take(1)))
    generate_images(generator, ex_input, ex_output, i, test=True)

我如何手动要求检查点管理器检索这个或那个特定的检查点文件，而不是.latest_checkpoint()？

编辑：自己解决，打开检查点文件夹中的 checkpoints.txt 文件，并将后缀号设置为要加载的任何检查点。

【问题讨论】：

标签： tensorflow

【解决方案1】：

您可以使用 checkpoints.restore() 方法恢复您偏好的检查点。例如，如果要在迭代 1000 处加载检查点，则编写：

checkpoint.restore('./test/model.ckpt-1000')

有关详细信息，请参阅此documentation。谢谢你。

【讨论】：