update

BUPT-GAMMA · Jun 26, 2024 · 4b6f187 · 4b6f187
1 parent a460dbf
commit 4b6f187
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gammagl/layers/conv/gat_conv.py b/gammagl/layers/conv/gat_conv.py
@@ -81,11 +81,11 @@ def __init__(self,
 
         init_weight = tlx.initializers.TruncatedNormal()
         self.w = tlx.nn.Parameter(
-            init_weight((in_channels, self.out_channels * self.heads)), name='linear_weight')
+            init_weight((in_channels, self.out_channels * self.heads)))
 
         initor = tlx.initializers.TruncatedNormal()
         self.att = tlx.nn.Parameter(
-            initor((1, self.heads, self.out_channels * 2)), name='att')
+            initor((1, self.heads, self.out_channels * 2)))
 
         self.leaky_relu = tlx.layers.LeakyReLU(negative_slope)
         self.dropout = tlx.layers.Dropout(self.dropout_rate)