使用自定义图层加载Keras模型的权重

Question

我正在尝试保存以下模型中的WGAN实现的模型和权重

https://github.com/keras-team/keras-contrib/blob/master/examples/improved_wgan.py。

但是，当我们执行load_weights（...）时，它会引发“轴与数组不匹配”错误。 我试图使用保存模型，使用自定义对象加载模型。但这也引发了错误。 当我们在Keras中有自定义损失和自定义图层时，有人可以帮助我正确保存权重吗？

Answer 1

如果有人需要解决方案，请将模型保存为yaml格式，然后h5中的权重对我有用！