Unverified Commit 861666fa authored by Xin Yao's avatar Xin Yao Committed by GitHub
Browse files

[Bugfix] Fix RGCN Example (#3785)

parent ec64581c
......@@ -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)
......
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