赞
踩
1问题描述:
出现:ValueError: Unknown layer:**(此处因为会有各种情况,不详细写出)
使用keras搭建模型,并成功训练和保存了模型结构和参数,但当load_model时,便出现了该问题,一开始的想法是使用model.to_json()方式保存模型结构,使用model.save_weights()方式保存权重参数,但加载模型结构时仍然出现此问题。
2原因与方案
翻墙到xx星球去寻求答案,并未找到本人一样的问题,但庆幸能看懂一点点论坛英文,参考了一下,发现问题解决了,神奇!!!,具体如下:
**原因:**之所以在加载模型结构时出现此问题,是因为搭建模型时,我们创建了一些“新”的层,指keras中不是默认存在的层,所以加载结构时报错。
**方案:**训练时保存权重(使用model.save_weights()方法);测试时直接导入搭建的模型代码(建议写成类或者函数封装,加载很方便),结构有了,下一步就是使用model.load_weights()来导入权重文件。
问题解决,祝好运!!!
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。