[Bugfix] Fix RGCN Example (#3785)

dmlc · Feb 28, 2022 · 861666f · 861666f
1 parent ec64581
commit 861666f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/pytorch/rgcn/entity_sample_multi_gpu.py b/examples/pytorch/rgcn/entity_sample_multi_gpu.py
@@ -97,6 +97,7 @@ def main(args, devices):
 
     # Create csr/coo/csc formats before launching training processes.
     # This avoids creating certain formats in each sub-process, which saves momory and CPU.
+    g = data[0]
     g.create_formats_()
 
     n_gpus = len(devices)