DGL-KE TransR Predict Error

[zezhongding]

When I predict using TransR in my dataset after training, it comes to the following errors.
traceback (most recent call last): File "/home4/intern/zzding3/anaconda3/envs/embed/lib/python3.9/multiprocessing/process.py"，Line315,inbootstrapself run (File "/home4/intern/zzding3/anaconda3/envs/embed/lib/python3.9/multiprocessing/process.py", line108,in runself. target(*self. args,**self，kwargs)File "/home4/intern/zzding3/.local/lib/python3.9/site-packages/dglke-0.1.0.dev0-py3.9.egg/dglke)models/pytorch/tensor models.py", line 133, in decorated functionraise exception .class(trace)IndexError: Traceback (most recent call last):File "/home4/intern/zzding3/. ocal/lib/python3.9/site-packages/dglke-0.1.0.dev0-py3.9.egg/dglkemodels/pytorch/tensor models .py，line 121,inqueue resultres = func(*args，**kwargs)File " /home4/intern/zzding3/.local/lib/python3.9/site-packages/dglke-0.1.0.dev0-py3.9.egg/dglketrain pytorch.py",line 265,in test mptest(args, model, test samplers, rank,mode, queue)File "/home4/intern/zzding3/. ocal/lib/python3.9/site-packages/dglke-0.1.0,dev0-py3.9,egg/dglke)train pytorch.py",line 241, in testmodel.forward test(pos g,neg g, logs, gpu id)File "/home4/intern/zzding3/.local/lib/python3.9/site-packages/dglke-0.1.0.dev0-py3.9.egg/dglke)models/general models .py",line 458,in forward testneg scores = selfpredict neg score(pos g, neg g,to device=cuda,File "/home4/intern/zzding3/.local/lib/python3.9/site-packages/dglke-0,1.0.dev0-py3.9.egg/dglke/models/general models,py", ine 383,in predict neg scoreneg head = selfentity emb(neg head ids, gpu id,trace)File "/home4/intern/zzding3/.local/lib/python3.9/site-packages/dglke-0.1.0.dev0-py3.9.egg/dglkemodels/pytorch/tensor models.py", line 292, incalls = self,emblidx]IndexError: index 62286 is out of bounds for dimension 0 with size 62286
Does someone know the reasons?