当前位置:   article > 正文

ValueError: Unknown layer:**_valueerror: unknown layer: layernormalization

valueerror: unknown layer: layernormalization

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

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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/499827
推荐阅读
相关标签
  

闽ICP备14008679号