Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "29b1325a5ae28fa8d7f459b372582287ffc571e5"
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
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
examples/pytorch/graphsage/model.py
View file @ 01a22144
...@@ -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