remove useless args

wang-chen · Feb 1, 2021 · 4437397 · 4437397
1 parent c9b39f5
commit 4437397
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gat.py b/gat.py
@@ -28,7 +28,7 @@ def forward(self, x):
 
 
 class GraphAttn(nn.Module):
-    def __init__(self, in_channels, out_channels, kernel_size, stride, padding, feat_len, alpha=0.2, dropout=0.2):
+    def __init__(self, in_channels, out_channels, kernel_size, stride, padding, feat_len, alpha=0.2):
         super().__init__()
         self.tran = nn.Conv2d(in_channels, out_channels, kernel_size, stride, padding, padding_mode='circular')
         self.att1 = nn.Linear(feat_len, 1, bias=False)