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

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

不懂是什么原因，反正在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