Skip to content
Permalink
Browse files

add num workers

  • Loading branch information
rusty1s committed Mar 26, 2020
1 parent 4d3808f commit cddd167ca54f7158dffdfeb874aea947e90b0657
Showing with 1 addition and 1 deletion.
  1. +1 −1 examples/graph_saint.py
@@ -16,7 +16,7 @@
loader = GraphSAINTRandomWalkSampler(data, batch_size=6000, walk_length=2,
num_steps=5, sample_coverage=1000,
save_dir=dataset.processed_dir,
num_workers=0)
num_workers=4)


class Net(torch.nn.Module):

0 comments on commit cddd167

Please sign in to comment.
You can’t perform that action at this time.