问题描述:
在实现自编码器的过程中,加入了自定义的采样层,训练完成之后,加载模型失败
解决方法:
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