Merge pull request #254 from RyushiAok/fix/mpnnlstm_example

Remove an extra argument in MPNNLSTM example
benedekrozemberczki · Oct 14, 2024 · 5fe0f13 · 5fe0f13
2 parents 97a6f00 + be51d94
commit 5fe0f13
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/recurrent/mpnnlstm_example.py b/examples/recurrent/mpnnlstm_example.py
@@ -20,7 +20,7 @@ def tqdm(iterable):
 class RecurrentGCN(torch.nn.Module):
     def __init__(self, node_features):
         super(RecurrentGCN, self).__init__()
-        self.recurrent = MPNNLSTM(node_features, 32, 32, 20, 1, 0.5)
+        self.recurrent = MPNNLSTM(node_features, 32, 20, 1, 0.5)
         self.linear = torch.nn.Linear(2*32 + node_features, 1)
 
     def forward(self, x, edge_index, edge_weight):