Skip to content

GitLab

"tests/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "55d49d4379007740af20629bb61aba9546c6b053"
Unverified Commit 367a3a34 authored by Junwen Yao's avatar Junwen Yao Committed by GitHub
Browse files

Fix torch import in example (#3372)

parent a04a8d06
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/graphsage/train_cv_multi_gpu.py
View file @ 367a3a34
...@@ -239,7 +239,7 @@ def run(proc_id, n_gpus, args, devices, data): ...@@ -239,7 +239,7 @@ def run(proc_id, n_gpus, args, devices, data):
# Create PyTorch DataLoader for constructing blocks # Create PyTorch DataLoader for constructing blocks
if n_gpus > 1: if n_gpus > 1:
dist_sampler = torch.utils.data.distributed.DistributedSampler(train_nid.numpy(), shuffle=True, drop_last=False) dist_sampler = th.utils.data.distributed.DistributedSampler(train_nid.numpy(), shuffle=True, drop_last=False)
dataloader = DataLoader( dataloader = DataLoader(
dataset=train_nid.numpy(), dataset=train_nid.numpy(),
batch_size=args.batch_size, batch_size=args.batch_size,
......
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