ValueError: Unknown layer: SampleLayer 常見問題1:模型中包含自定義的layer,加載模型時出錯

問題描述:

在實現自編碼器的過程中,加入了自定義的採樣層,訓練完成之後,加載模型失敗

解決方法: 

load_model方法提供了custom_objects的參數,該參數接受一個dict型的輸入,key值爲model.summary()中顯示的layer的名字,value的值爲自定義的layer類的類名

from model import SampleLayer

encoder = load_model(encoderpath, custom_objects={'SampleLayer':SampleLayer})

此外,keras還提供另外兩種解決方案:https://github.com/keras-team/keras/issues/8612

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章