如何在Tf SLIM中加载保存的检查点

Question

我正在使用TF-slim训练自己的数据集。 根据该教程，在finetune_inception_v1_on_flowers.sh中，我将检查点路径设置为./inception_resnet_v2.ckpt，并在另一个文件夹中获得许多新的检查点，最新文件为model.ckpt-332331.data-00000-of -00001，model.ckpt-332331.index，model.ckpt-332331.meta。

现在，我想将学习率更改为精确到最新的检查点，但是我不知道如何将它们加载到finetune_inception_v1_on_flowers.sh中，因为有3个文件而不是1个。

firstly, I tried to copy the  model.ckpt-332331.data-00000-of-00001 and set the  checkpoint path as ./model.ckpt-332331.data-00000-of-00001 ,but there is the error:

NotFoundError（请参阅上面的回溯）：密钥InceptionResnetV2 / AuxLogits / Conv2d_1b_1x1 / weights / Adam在检查点[[Node：save / RestoreV2_4 = RestoreV2 [dtypes = [DT_FLOAT]，_device =“ / job：localhost / replica：0 /任务：0 / cpu：0“]（_ recv_save / Const_0，保存/恢复V2_4 /张量名称，保存/恢复V2_4 / shape_and_slices）]]

其次，我从日志目录中删除了文件，那么错误是：

DataLossError（请参阅上面的回溯）：无法打开表文件/media/re/7A308F7E308F405D/xsj/checkpoints/model.ckpt-332331.data-00000-of-00001：数据丢失：不是sstable（错误的魔术数字）：也许您的文件采用其他文件格式，并且需要使用其他还原操作符？ [[节点：保存/还原V2_5 = RestoreV2 [dtypes = [DT_FLOAT]，_ device =“ / job：localhost /副本0 /任务：0 / cpu：0”]] __ recv_save / Const_0，保存/还原V2_5 / tensor_names，保存/ RestoreV2_5 / shape_and_slices）]]

谁能帮助我解决问题？ 非常感谢你！

Answer 1

如https://www.tensorflow.org/guide/saved_model中所述

没有一个名为/tmp/model.ckpt的物理文件。 它是为检查点创建的文件名的前缀。 用户仅与前缀交互，而不与物理检查点文件交互。

因此，您只需要将检查点路径设置为./model.ckpt-332331

Answer 2

进行微调时，只需指定目录（$ CHECKPOINT_PATH）而不是任何特定文件，它将自动获取路径。