当前位置: 代码迷 >> 综合 >> ValueError: Unknown layer:**
  详细解决方案

ValueError: Unknown layer:**

热度:67   发布时间:2023-11-21 01:43:36.0

1问题描述:
出现:ValueError: Unknown layer:**(此处因为会有各种情况,不详细写出)
使用keras搭建模型,并成功训练和保存了模型结构和参数,但当load_model时,便出现了该问题,一开始的想法是使用model.to_json()方式保存模型结构,使用model.save_weights()方式保存权重参数,但加载模型结构时仍然出现此问题。

2原因与方案
翻墙到xx星球去寻求答案,并未找到本人一样的问题,但庆幸能看懂一点点论坛英文,参考了一下,发现问题解决了,神奇!!!,具体如下:
**原因:**之所以在加载模型结构时出现此问题,是因为搭建模型时,我们创建了一些“新”的层,指keras中不是默认存在的层,所以加载结构时报错。
**方案:**训练时保存权重(使用model.save_weights()方法);测试时直接导入搭建的模型代码(建议写成类或者函数封装,加载很方便),结构有了,下一步就是使用model.load_weights()来导入权重文件。
问题解决,祝好运!!!

  相关解决方案