ValueError: Unknown layer: SampleLayer 常见问题1:模型中包含自定义的layer,加载模型时出错
问题描述:在实现自编码器的过程中,加入了自定义的采样层,训练完成之后,加载模型失败解决方法:load_model方法提供了custom_objects的参数,该参数接受一个dict型的输入,key值为model.summary()中显示的layer的名字,value的值为自定义的layer类的类名from model import SampleLayerencoder = load_model(encoderpath, custom_objects={'SampleLayer':.
原创
2020-05-15 10:52:29 ·
1884 阅读 ·
0 评论