Unverified Commit 01a22144 authored by nv-dlasalle's avatar nv-dlasalle Committed by GitHub
Browse files

Enable faster validation for pytorch graphsage example (#3361)

parent bc5cac44
......@@ -58,8 +58,9 @@ class SAGE(nn.Module):
g,
th.arange(g.num_nodes()).to(g.device),
sampler,
device=device if num_workers == 0 else None,
batch_size=batch_size,
shuffle=True,
shuffle=False,
drop_last=False,
num_workers=num_workers)
......
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