1

我正在使用 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（请参阅上面的回溯）：检查点中未找到 Key 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，保存/RestoreV2_4/tensor_names，保存/RestoreV2_4/shape_and_slices)]]

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

DataLossError（有关回溯，请参见上文）：无法打开表文件 /media/re/7A308F7E308F405D/xsj/checkpoints/model.ckpt-332331.data-00000-of-00001：数据丢失：不是 sstable（坏幻数）：也许您的文件是不同的文件格式，您需要使用不同的还原操作符？[[节点：save/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/恢复V2_5/shape_and_slices)]]

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

4

2 回答 2

2

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

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

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

于 2019-01-04T03:01:07.210 回答
0

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

于 2018-07-02T17:29:31.210 回答