僅權重儲存 & 載入
save_weights 方法
Model.save_weights(filepath, overwrite=True)
將所有層權重儲存到 .weights.h5 檔案。
參數
- filepath：
str或pathlib.Path物件。模型儲存路徑。必須以.weights.h5結尾。 - overwrite：是否應覆寫目標位置的任何現有模型，或透過互動式提示詢問使用者。
load_weights 方法
Model.load_weights(filepath, skip_mismatch=False, **kwargs)
從透過 save_weights() 儲存的檔案載入權重。
權重根據網路拓撲載入。這表示架構應與儲存權重時相同。請注意，不具有權重的層不會納入拓撲排序考量，因此新增或移除不具有權重的層是可以的。
部分權重載入
如果您修改了模型，例如新增了新的層（具有權重）或變更了層權重的形狀，您可以選擇忽略錯誤並繼續載入，方法是設定 skip_mismatch=True。在這種情況下，任何權重不符的層都會被略過。對於每個略過的層，都會顯示警告。
參數
- filepath：字串，權重檔案載入路徑。它可以是
.weights.h5檔案或舊版.h5權重檔案。 - skip_mismatch：布林值，是否跳過載入權重數量不符或權重形狀不符的層。