Skip to content

Commit

Permalink
neighbor sample from [25, 10] to [10, 5]
Browse files Browse the repository at this point in the history
  • Loading branch information
xumwen committed Mar 30, 2020
1 parent 91f52fc commit fc1f113
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion gcn.py
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,7 @@ def forward(self, X, edge_index, edge_weight):
# and constructs sampled subgraphs (use cpu for no CUDA version)
out = torch.zeros(sz[0], sz[1], self.out_channels, device=X.device)
graph_data = Data(edge_index=edge_index, num_nodes=sz[1]).to('cpu')
loader = NeighborSampler(graph_data, size=[25, 10], num_hops=2, batch_size=100,
loader = NeighborSampler(graph_data, size=[10, 5], num_hops=2, batch_size=100,
shuffle=True, add_self_loops=False)

for data_flow in loader():
Expand Down

0 comments on commit fc1f113

Please sign in to comment.