Skip to content

GitLab

Unverified Commit a0bf5daa authored by Tong He's avatar Tong He Committed by GitHub
Browse files

fix (#3903) 


Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent c380f0b5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/hilander/models/graphconv.py
View file @ a0bf5daa
...@@ -19,7 +19,7 @@ class GraphConvLayer(nn.Module): ...@@ -19,7 +19,7 @@ class GraphConvLayer(nn.Module):
srcfeat, dstfeat = feat srcfeat, dstfeat = feat
else: else:
srcfeat = feat srcfeat = feat
dstfeat = feat[:graph.num_dst_nodes()] dstfeat = feat[:bipartite.num_dst_nodes()]
graph = bipartite.local_var() graph = bipartite.local_var()
graph.srcdata['h'] = srcfeat graph.srcdata['h'] = srcfeat
......
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