Python 具有自定义图层的keras模型的负载权重

我正在尝试保存模型和权重，以便在中找到WGAN实现

但是，当我们执行load_权重（…）时，它会抛出“轴与数组不匹配”错误。 我尝试使用保存模型，加载带有自定义对象的模型。但这也带来了错误。

---

当我们在Keras中有自定义损耗和自定义层时，有人能帮助我正确保存权重吗？

如果有人需要解决方案，以yaml格式保存模型，然后以h5格式保存权重对我有效