fix the bug in HeteroSAGEConv

snap-stanford · Jun 27, 2020 · 02eaa68 · 02eaa68
1 parent 48d9ade
commit 02eaa68
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/deepsnap/hetero_gnn.py b/deepsnap/hetero_gnn.py
@@ -26,9 +26,9 @@ def __init__(self, in_channels_neigh, out_channels, in_channels_self=None):
         else:
             self.in_channels_self = in_channels_self
         self.out_channels = out_channels
-        self.lin_neigh = nn.Linear(in_channels_neigh, out_channels)
-        self.lin_self = nn.Linear(in_channels_self, out_channels)
-        self.lin_update = nn.Linear(out_channels * 2, out_channels)
+        self.lin_neigh = nn.Linear(self.in_channels_neigh, self.out_channels)
+        self.lin_self = nn.Linear(self.in_channels_self, self.out_channels)
+        self.lin_update = nn.Linear(self.out_channels * 2, self.out_channels)
 
     def forward(self, node_feature_neigh, node_feature_self,
                 edge_index, edge_weight=None, size=None,