Tensorflow 编写模块化Keras代码的推荐方法是什么？

我正在努力编写可组合和模块化的Keras代码，我想知道我是否只是使用了错误的API？我想要的是：

将keras层和自定义层包裹在一起

易于重复使用，理想情况下就像任何内置Keras层一样

保存和加载也必须起作用

• 最好是在没有代码的情况下制作它

我尝试子类化

模型

，但无法保存这些模型，除非：

序列化由子类网络的作者实现

我可以将代码更改为从

层继承

，但是权重似乎没有训练，即使它运行时没有任何错误

---

那么，我的选择是什么？这里的示例链接最有帮助。

我认为您需要为您的

自定义层

在从

层

类中编写一个类。你可以查看这个网站@Faruk这个问题是包含其他现有层的自定义层不起作用——看起来内部层不训练。您知道有任何自定义层嵌套并可以训练的工作示例吗？遗憾的是，不知道。也许这个搜索会给您一些线索。我认为您需要在RETING from

layer

类中为您的

自定义层编写一个类。你可以查看这个网站@Faruk这个问题是包含其他现有层的自定义层不起作用——看起来内部层不训练。您知道有任何自定义层嵌套并可以训练的工作示例吗？遗憾的是，不知道。也许这个搜索会给您一些线索。