ADAMANCY361

referce:https://blog.csdn.net/potato_uncle/article/details/109699651 self.learnable_factor = self.add_weight(shape=(1, 1, 1, 512), dtype=tf.float32, initializer=tf.keras.initializers.Ones(), trainable=True) change to : self.learnable_factor = self.add_weight(name='w',shape=(1, 1, 1, 512), dtype=tf.float32, initializer=tf.keras.initializers.Ones(), trainable=True)