問題描述:
在實現自編碼器的過程中,加入了自定義的採樣層,訓練完成之後,加載模型失敗
解決方法:
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