Skip to content

GitLab

Unverified Commit 714f2e1c authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

fix node id dtype (#674)

parent f8feee1c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/rgcn/link_predict.py
View file @ 714f2e1c
...@@ -148,7 +148,7 @@ def main(args): ...@@ -148,7 +148,7 @@ def main(args):
print("Done edge sampling") print("Done edge sampling")
# set node/edge feature # set node/edge feature
node_id = torch.from_numpy(node_id).view(-1, 1) node_id = torch.from_numpy(node_id).view(-1, 1).long()
edge_type = torch.from_numpy(edge_type) edge_type = torch.from_numpy(edge_type)
node_norm = torch.from_numpy(node_norm).view(-1, 1) node_norm = torch.from_numpy(node_norm).view(-1, 1)
data, labels = torch.from_numpy(data), torch.from_numpy(labels) data, labels = torch.from_numpy(data), torch.from_numpy(labels)
......
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