alibaba / graph-learn

An Industrial Graph Neural Network Framework
Apache License 2.0
1.28k stars 267 forks source link

GCN模型Pytorch样例代码bug #230

Closed lxx1818 closed 1 year ago

lxx1818 commented 1 year ago

bug目录:examples/pytorch/gcn/gcn.py

examples/pytorch/gcn/gcn.py的GCN的模型样例代码,在代码第40行出现了一个bug。

38 for i in range(depth): 
39      input_dim = input_dim if i == 0 else hidden_dim
40      output_dim = output_dim if i == depth - 1 else hidden_dim
41      self.layers.append(GCNConv(input_dim, output_dim))

运行过程中,第一轮迭代后output_dim数值将被改写为hidden_dim,导致后续模型输出dim不对——目前代码虽然不会报错,但是后导致模型训练效果不佳。