Unverified Commit a90296aa authored by Da Zheng's avatar Da Zheng Committed by GitHub
Browse files

fix. (#2907)


Co-authored-by: default avatarZheng <dzzhen@3c22fba32af5.ant.amazon.com>
parent e156331b
...@@ -100,7 +100,7 @@ the mapping for each node type; the second dictionary contains the mapping for e ...@@ -100,7 +100,7 @@ the mapping for each node type; the second dictionary contains the mapping for e
return_mapping=True) return_mapping=True)
# Let's assume that node_emb is saved from the distributed training. # Let's assume that node_emb is saved from the distributed training.
orig_node_emb = th.zeros(node_emb.shape, dtype=node_emb.dtype) orig_node_emb = th.zeros(node_emb.shape, dtype=node_emb.dtype)
orig_labels[node_map] = node_emb orig_node_emb[node_map] = node_emb
7.1.1 Distributed partitioning 7.1.1 Distributed partitioning
......
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