"IndexError: index out of range in self" when I train on my own knowledge graph. Can anyone help me with this? Thanks a lot!

[YoannCheung]

2023-11-12 21:50:47,687 INFO Model: RotatE 2023-11-12 21:50:47,687 INFO Data Path: data/DiseaseKG 2023-11-12 21:50:47,688 INFO #entity: 43465 2023-11-12 21:50:47,688 INFO #relation: 10 2023-11-12 21:50:47,966 INFO #train: 291112 2023-11-12 21:50:47,966 INFO #valid: 0 2023-11-12 21:50:47,966 INFO #test: 0 2023-11-12 21:50:48,252 INFO Model Parameter Configuration: 2023-11-12 21:50:48,252 INFO Parameter gamma: torch.Size([1]), require_grad = False 2023-11-12 21:50:48,252 INFO Parameter embedding_range: torch.Size([1]), require_grad = False 2023-11-12 21:50:48,252 INFO Parameter entity_embedding: torch.Size([43465, 1000]), require_grad = True 2023-11-12 21:50:48,252 INFO Parameter relation_embedding: torch.Size([10, 500]), require_grad = True 2023-11-12 21:50:49,822 INFO Ramdomly Initializing RotatE Model... 2023-11-12 21:50:49,822 INFO Start Training... 2023-11-12 21:50:49,822 INFO init_step = 0 2023-11-12 21:50:49,822 INFO batch_size = 512 2023-11-12 21:50:49,822 INFO negative_adversarial_sampling = 1 2023-11-12 21:50:49,822 INFO hidden_dim = 500 2023-11-12 21:50:49,822 INFO gamma = 12.000000 2023-11-12 21:50:49,822 INFO negative_adversarial_sampling = True 2023-11-12 21:50:49,822 INFO adversarial_temperature = 0.500000 2023-11-12 21:50:49,822 INFO learning_rate = 0 Traceback (most recent call last): File "/home/zyx/Projects/KGE/KnowledgeGraphEmbedding-master/codes/run.py", line 361, in main(parse_args()) File "/home/zyx/Projects/KGE/KnowledgeGraphEmbedding-master/codes/run.py", line 305, in main log = kge_model.train_step(kge_model, optimizer, train_iterator, args) File "/home/zyx/Projects/KGE/KnowledgeGraphEmbedding-master/codes/model.py", line 267, in train_step negative_score = model((positive_sample, negative_sample), mode=mode) File "/home/zyx/anaconda3/envs/KGE/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl return forward_call(*input, **kwargs) File "/home/zyx/Projects/KGE/KnowledgeGraphEmbedding-master/codes/model.py", line 129, in forward head = torch.index_select( IndexError: index out of range in self