meettyj / NOSMOG

Code & data for ICLR'23 Spotlight paper "Learning MLPs on Graphs: A Unified View of Effectiveness, Robustness, and Efficiency".
29 stars 11 forks source link

Error in inductive settiing on product dataset #4

Open scottjiao opened 9 months ago

scottjiao commented 9 months ago

Meet error while running following command: python train_student.py --exp_setting ind --teacher "SAGE" --student "MLP" --dataset "ogbn-products" --num_exp 10 --max_epoch 200 --patience 30 --device 0 --device 1

File "/data/zhou/NOSMOG-master/train_student.py", line 468, in run position_feature_obs[idx_cur_node_id] = torch.tensor(temp_position_feature, dtype=torch.float32) IndexError: index 2396430 is out of bounds for dimension 0 with size 2006411

It looks like the code try to access unobserved test_ind idx in a tensor where only observed idx are contained.