Skip to content

GitLab

Unverified Commit 421c3622 authored by Jinjing Zhou's avatar Jinjing Zhou Committed by GitHub
Browse files

Update train_dist.py (#3594) 


Co-authored-by: default avatarQuan (Andy) Gan <coin2028@hotmail.com>
parent 5eca59d8
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/graphsage/experimental/train_dist.py
View file @ 421c3622
...@@ -172,7 +172,7 @@ def pad_data(nids): ...@@ -172,7 +172,7 @@ def pad_data(nids):
if max_num_nodes > nids_length: if max_num_nodes > nids_length:
pad_size = max_num_nodes % nids_length pad_size = max_num_nodes % nids_length
repeat_size = max_num_nodes // nids_length repeat_size = max_num_nodes // nids_length
new_nids = th.cat([nids for _ in repeat_size] + [nids[:pad_size]], axis=0) new_nids = th.cat([nids for _ in range(repeat_size)] + [nids[:pad_size]], axis=0)
print("Pad nids from {} to {}".format(nids_length, max_num_nodes)) print("Pad nids from {} to {}".format(nids_length, max_num_nodes))
else: else:
new_nids = nids new_nids = nids
......
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