TypeError: init() got an unexpected keyword argument ‘name‘

最新推荐文章于 2024-12-12 20:22:45 发布

原创最新推荐文章于 2024-12-12 20:22:45 发布 · 2.4k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode

debug 专栏收录该内容

57 篇文章

订阅专栏

本文介绍了一种常见的Keras自定义层实现错误及解决方案。通过调整自定义层类的初始化方法，加入**kwargs参数，可以成功避免错误并使自定义层正常工作。

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

When I wrote a custom class of Keras, I met this error.

Solution:

change from the snippet below

class custconv2d(keras.layers.Layer):
    def __init__(self):
        super(custconv2d, self).__init__()
        self.k = self.add_weight(shape=(1, ), initializer="random_normal", trainable=True, name='k1')

#add **kwargs in the __init__()
class custconv2d(keras.layers.Layer):
    def __init__(self, **kwargs):
        super(custconv2d, self).__init__(**kwargs)
        self.k = self.add_weight(shape=(1, ), initializer="random_normal", trainable=True, name='k1')