[pycharm] NotImplementedError: Layer XX has arguments in `init` and therefore must override `get

Bing_Zoey

于 2023-05-31 23:30:38 发布

阅读量435

点赞数

分类专栏： MI-EEG 文章标签： python 开发语言

本文链接：https://blog.youkuaiyun.com/diamondssb/article/details/130979042

版权

MI-EEG 专栏收录该内容

13 篇文章

订阅专栏

当在Keras中定义自定义层时，如果`__init__`中有参数，需要覆盖`get_config`方法以保存层配置。文章提到的问题是由于没有实现`get_config`导致的错误，解决方案是在类中添加该方法并返回相关配置信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考(1条消息) 解决NotImplementedError: Layer XX has arguments in `__init__` and therefore must override `get_config`_sinysama的博客-优快云博客

不懂是什么原因，反正在class后面加上get_config就行

我的class整体代码如下

class AttentionBlock(layers.Layer):
    def __init__(self, num_heads, ff_dim, rate):
        super(AttentionBlock, self).__init__()
        self.num_heads = num_heads
        self.ff_dim = ff_dim
        self.rate = rate
        ...

    def call(self, inputs, training):
        ...
    def get_config(self):
        config = super().get_config().copy()
        config.update({
            'num_heads' : self.num_heads,
            'ff_dim' : self.ff_dim,
            'rate' : self.rate,
        })
        return config