dimension problem

[ICDI0906]

Hello~
Recently, I am reading your paper carefully. In model EGCU, I'm confused about how the dimension of the parameter matrix change, can you explain it?
looking forward to your reply.

[benedekrozemberczki]

I would also like to raise this issue, the current code release is not sufficient for replication.

[dgx11678832]

Traceback (most recent call last):
File "/home/dgx/PycharmProjects/EvolveGCN-master/test_args.py", line 237, in
trainer.train()
File "/home/dgx/PycharmProjects/EvolveGCN-master/trainer.py", line 63, in train
eval_train, nodes_embs = self.run_epoch(self.splitter.train, e, 'TRAIN', grad = True)
File "/home/dgx/PycharmProjects/EvolveGCN-master/trainer.py", line 104, in run_epoch
loss = self.comp_loss(predictions,s.label_sp['vals'])
File "/home/dgx/anaconda3/envs/EvolveGCN-master/lib/python3.6/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/dgx/PycharmProjects/EvolveGCN-master/Cross_Entropy.py", line 47, in forward
alpha = self.weights(labels)[labels].view(-1,1)
IndexError: index 2 is out of bounds for dimension 0 with size

Have you encountered this problem?