Skip to content

GitLab

Unverified Commit d34cbf31 authored by Mufei Li's avatar Mufei Li Committed by GitHub
Browse files

Update train_sampling_unsupervised.py (#1973)

parent 5454471f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/graphsage/train_sampling_unsupervised.py
View file @ d34cbf31
...@@ -350,7 +350,7 @@ def main(args, devices): ...@@ -350,7 +350,7 @@ def main(args, devices):
n_gpus = len(devices) n_gpus = len(devices)
if devices[0] == -1: if devices[0] == -1:
run(0, 0, args, ['cpu'], data) run(0, 0, args, ['cpu'], data)
if n_gpus == 1: elif n_gpus == 1:
run(0, n_gpus, args, devices, data) run(0, n_gpus, args, devices, data)
else: else:
procs = [] procs = []
......
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