"git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "5f04fc2b31410a8801d729207fd8f28ce233bd92"
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): ...@@ -58,8 +58,9 @@ class SAGE(nn.Module):
g, g,
th.arange(g.num_nodes()).to(g.device), th.arange(g.num_nodes()).to(g.device),
sampler, sampler,
device=device if num_workers == 0 else None,
batch_size=batch_size, batch_size=batch_size,
shuffle=True, shuffle=False,
drop_last=False, drop_last=False,
num_workers=num_workers) 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