TensorFlow add_weight() got multiple values for argument 'name'

我自定义了一继承Layer的卷积层ComplexConv2D(sf, filsize, name='conv1', **convArgs)
卷积层里卷积核部分的代码为:

        self.kernel = self.add_weight(
            self.kernel_shape,
            name='kernel',
            initializer=kern_init,
            regularizer=self.kernel_regularizer,
            constraint=self.kernel_constraint

在我调用这个卷积层的时候报错,

TensorFlow add_weight() got multiple values for argument 'name'_第1张图片
根据这种类型的报错我上网查阅,根据一些说法https://blog.csdn.net/york1996/article/details/84071974?depth_1-utm_source=distribute.pc_relevant.none-task&utm_source=distribute.pc_relevant.none-task

说的是给一些参数多个赋值,如

def ss(a,b):
   pass
ss(1,2,b=1)

解决

后来把add_weight中的shape改成shape = self.kernel_shape就没问题了

你可能感兴趣的:(tensorflow)