Commit eec1d4c7 authored by rusty1s's avatar rusty1s
Browse files

typos

parent a44bec9e
...@@ -26,11 +26,11 @@ class GNN(ScalableGNN): ...@@ -26,11 +26,11 @@ class GNN(ScalableGNN):
self.convs.append(GCNConv(hidden_channels, hidden_channels)) self.convs.append(GCNConv(hidden_channels, hidden_channels))
self.convs.append(GCNConv(hidden_channels, out_channels)) self.convs.append(GCNConv(hidden_channels, out_channels))
def forward(self, x, adj, batch_size, n_id): def forward(self, x, adj_t, batch_size, n_id):
for conv, history in zip(self.convs[:-1], self.histories): for conv, history in zip(self.convs[:-1], self.histories):
x = conv(x, adj).relu_() x = conv(x, adj_t).relu_()
x = self.push_and_pull(history, x, batch_size, n_id) x = self.push_and_pull(history, x, batch_size, n_id)
return self.convs[-1](x, adj return self.convs[-1](x, adj_t)
``` ```
## Installation ## Installation
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment