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

【问题标题】：how to load saved checkpoint in Tf SLIM如何在 Tf SLIM 中加载保存的检查点
【发布时间】：2018-08-13 06:02:57
【问题描述】：

我正在使用 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个文件而不是一个。

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 未在检查点 [[节点：save/RestoreV2_4 = RestoreV2[dtypes=[DT_FLOAT]， _device="/job:localhost/replica:0/task:0/cpu:0"](_recv_save/Const_0, save/RestoreV2_4/tensor_names, save/RestoreV2_4/shape_and_slices)]]

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

DataLossError（参见上面的回溯）：无法打开表文件 /media/re/7A308F7E308F405D/xsj/checkpoints/model.ckpt-332331.data-00000-of-00001：数据丢失：不是 sstable（坏幻数）：也许你的文件在不同的文件格式，您需要使用不同的还原操作员？ [[节点：保存/RestoreV2_5 = RestoreV2[dtypes=[DT_FLOAT]， _device="/job:localhost/replica:0/task:0/cpu:0"](_recv_save/Const_0, save/RestoreV2_5/tensor_names, save/RestoreV2_5/shape_and_slices)]]

谁能帮我解决这个问题？非常感谢！

【问题讨论】：

标签： python tensorflow model checkpoint tf-slim

【解决方案1】：

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

【讨论】：

【解决方案2】：

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

没有名为 /tmp/model.ckpt 的物理文件。这是前缀为检查点创建的文件名。用户只与前缀而不是物理检查点文件。

所以你只需要将检查点路径设置为./model.ckpt-332331

【讨论】：