如何在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）而不是任何特定文件，它將自動獲取路徑。