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

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值