Merge pull request #79 from natsukium/patch-1

Fix argument of Set2set in example
pyg-team · Jan 23, 2019 · 1f97601 · 1f97601
2 parents 407a074 + 162966f
commit 1f97601
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/qm9_nn_conv.py b/examples/qm9_nn_conv.py
@@ -75,7 +75,7 @@ def __init__(self):
         nn = Sequential(Linear(5, 128), ReLU(), Linear(128, dim * dim))
         self.conv = NNConv(dim, dim, nn, root_weight=False)
         self.gru = GRU(dim, dim, batch_first=True)
-        self.set2set = Set2Set(dim, dim, processing_steps=3)
+        self.set2set = Set2Set(dim, processing_steps=3)
         self.fc1 = torch.nn.Linear(2 * dim, dim)
         self.fc2 = torch.nn.Linear(dim, 1)