我應該使用哪個檢查點 select 繼續進行 Object 檢測訓練

Question

我開始訓練直到 ckpt-7 然后我停止訓練。 然后我再次開始訓練，但是在我更改了我的 model 上的微調檢查點的管道配置之前。 我寫了最新的檢查點，我改變了它的目錄。 在停止訓練之前，我的損失 function 大約為 0.899。 當我繼續訓練但它開始到步驟 100 和我的損失函數 15.009。

在停止之前如何繼續 model？ 我應該怎么辦？

我在 Colab 中使用 centernet model。 請解釋我是該主題的新手。

Answer 1

我可以理解您的問題，即您無法在停止的地方恢復培訓。

實際上，隨着 TF2 中的更新，我們無需更改 pipeline.config 中的微調檢查點參數。 重新運行指向存儲檢查點的同一 model_dir 的相同訓練腳本。

在 model_dir 中創建的檢查點的幫助下，TF2 將自動理解並從訓練停止的位置恢復。