williamleif/graphsage-simple

so many bugs when gcn=False ?

xuChenSJTU opened this issue · 3 comments

image
image

It seems that the aggregator does not work and the settings here is also strange
image

And these bugs only appear when in GPU training mode.

I have encountered the same problem, how did you solve it?

These settings work for me:

agg1 = MeanAggregator(features, cuda=False)
enc1 = Encoder(features, 1433, 128, adj_lists, agg1, gcn=False, cuda=False)
agg2 = MeanAggregator(lambda nodes : enc1(nodes).t(), cuda=False)
enc2 = Encoder(lambda nodes : enc1(nodes).t(), enc1.embed_dim, 128, adj_lists, agg2,
base_model=enc1, gcn=False, cuda=False)

My results:
Test score: 0.856551413074322
Test accuracy: 0.8379999995231628