Skip to content

GitLab

Commit 44405d11 authored by rusty1s's avatar rusty1s
Browse files

update

parent db97023e
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
README.md
View file @ 44405d11
...@@ -18,7 +18,6 @@ from torch_geometric.nn import SAGEConv ...@@ -18,7 +18,6 @@ from torch_geometric.nn import SAGEConv
from torch_geometric_autoscale import ScalableGNN from torch_geometric_autoscale import ScalableGNN
from torch_geometric_autoscale import metis, permute, SubgraphLoader from torch_geometric_autoscale import metis, permute, SubgraphLoader
class GNN(ScalableGNN): class GNN(ScalableGNN):
def __init__(self, num_nodes, in_channels, hidden_channels, out_channels, num_layers): def __init__(self, num_nodes, in_channels, hidden_channels, out_channels, num_layers):
# pool_size determines the number of pinned CPU buffers # pool_size determines the number of pinned CPU buffers
...@@ -41,8 +40,6 @@ class GNN(ScalableGNN): ...@@ -41,8 +40,6 @@ class GNN(ScalableGNN):
x = self.push_and_pull(history, x, *args) x = self.push_and_pull(history, x, *args)
return self.convs[-1](x, adj_t) return self.convs[-1](x, adj_t)
perm, ptr = metis(data.adj_t, num_parts=40, log=True) perm, ptr = metis(data.adj_t, num_parts=40, log=True)
data = permute(data, perm, log=True) data = permute(data, perm, log=True)
loader = SubgraphLoader(data, ptr, batch_size=10, shuffle=True) loader = SubgraphLoader(data, ptr, batch_size=10, shuffle=True)
......
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