HazyResearch / hgcn

Hyperbolic Graph Convolutional Networks in PyTorch.
602 stars 109 forks source link

Mistake in "hyperboloid.py" #45

Open Zhaoyang-Chu opened 2 years ago

Zhaoyang-Chu commented 2 years ago

When I ran the model with the following command: python train.py --task nc --dataset disease_nc --model HGCN --dim 16 --lr 0.01 --dim 16 --num-layers 2 --act relu --bias 1 --dropout 0 --weight-decay 0 --manifold Hyperboloid --log-freq 5 --cuda 0 --use-att 1 --local-agg 1

I got the following error: image IndexError: too many indices for tensor of dimension 1

I felt confused about this problem. Please help me.