加载权重 keras model 自定义层
[英]Load weights of keras model with custom layers
问题描述
我正在尝试保存 model 和 WGAN 实现的权重
https://github.com/keras-team/keras-contrib/blob/master/examples/improved_wgan.py 。
但是，当我们执行 load_weights(...) 时，它会抛出“轴不匹配数组”错误。 我尝试使用保存 model，加载带有自定义对象的 model。 但这也会引发错误。 当我们在 Keras 中有自定义损失和自定义层时，任何人都可以帮助我正确保存权重吗？
1 个解决方案
解决方案1
0 2020-09-21 16:56:52
如果有人需要解决方案，请将 model 保存为 yaml 格式，然后 h5 中的权重对我有用！
如何在Keras中加载卷积神经网络前几层的权重并删除预训练的model？
[英]How to load the weights of the first few layers of Convolutional Neural Network in Keras and delete the pre-trained model?
从keras中的预训练模型加载权重时的层误差
[英]Error in layers while loading weights from a pretrained model in keras for finetuning
Model 子类化忽略附加到 Python 列表的 Keras 层的权重
[英]Model subclassing ignores the weights of the Keras layers that appended to a Python list
如何可视化 keras model 中所有层的学习训练权重？
[英]How to visualize the learned trained weights in a keras model for all of its layers?
声明:本站的技术帖子网页，遵循CC BY-SA 4.0协议，如果您需要转载，请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.