Update train_sampling_unsupervised.py (#1973)

d34cbf31 · Mufei Li · GitHub · 5454471f · d34cbf31
Unverified Commit d34cbf31 authored Aug 08, 2020 by Mufei Li Committed by GitHub Aug 08, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/pytorch/graphsage/train_sampling_unsupervised.py examples/pytorch/graphsage/train_sampling_unsupervised.py +1 -1

No files found.
--- a/examples/pytorch/graphsage/train_sampling_unsupervised.py
+++ b/examples/pytorch/graphsage/train_sampling_unsupervised.py
@@ -350,7 +350,7 @@ def main(args, devices):
    n_gpus = len(devices)
    if devices[0] == -1:
        run(0, 0, args, ['cpu'], data)
-    if n_gpus == 1:
+    elif n_gpus == 1:
        run(0, n_gpus, args, devices, data)
    else:
        procs = []