我如何了解哪些層被凍結,對 Tensorflow Model Zoo 2 中的檢測模型進行微調?我已經成功設定了fine_tune_checkpoint的路徑,fine_tune_checkpoint_type: detection并且在檔案proto中我已經閱讀了“檢測”的意思
// 2. "detection": Restores the entire feature extractor.
The only parts of the full detection model that are not restored are the box and class prediction heads.
This option is typically used when you want to use a pre-trained detection model
and train on a new dataset or task which requires different box and class prediction heads.
我真的不明白這是什么意思。在這種情況下,恢復意味著凍結?
uj5u.com熱心網友回復:
據我了解,目前 Tensorflow 2 物件檢測在從微調檢查點進行訓練時不會凍結任何層。這里報告了一個問題,支持在管道配置中指定要凍結的層。如果查看訓練步驟函式,您可以看到在訓練期間應用梯度時使用了所有可訓練變數。
此處恢復意味著模型權重從檢查點復制以用作訓練的起點。凍結意味著在訓練期間權重不會改變(即沒有應用梯度)。
轉載請註明出處,本文鏈接:https://www.uj5u.com/gongcheng/352528.html
