twchen / lessr

Handling Information Loss of Graph Neural Networks for Session-based Recommendation
MIT License
66 stars 19 forks source link

Issue the request:RuntimeError #7

Closed lypingk closed 2 years ago

lypingk commented 2 years ago

Hi, do you have the following problem? Now the version is corresponding to the same, I don't know where the problem appears, is there any friend who can help solve it? Traceback (most recent call last): File "E:/WorkSpace2--pycharm/lessr-master/main.py", line 85, in runner.train(args.epochs, args.log_interval) File "E:\WorkSpace2--pycharm\lessr-master\utils\train.py", line 105, in train logits = self.model(inputs) File "E:\requireSoft\anaconda38\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl result = self.forward(input, *kwargs) File "E:\WorkSpace2--pycharm\lessr-master\lessr.py", line 171, in forward feat = self.embedding(iid) File "E:\requireSoft\anaconda38\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl result = self.forward(input, **kwargs) File "E:\requireSoft\anaconda38\lib\site-packages\torch\nn\modules\sparse.py", line 124, in forward return F.embedding( File "E:\requireSoft\anaconda38\lib\site-packages\torch\nn\functional.py", line 1813, in embedding _no_grad_embeddingrenorm(weight, input, max_norm, norm_type) File "E:\requireSoft\anaconda38\lib\site-packages\torch\nn\functional.py", line 1733, in _no_grad_embeddingrenorm torch.embeddingrenorm(weight, input, max_norm, norm_type) RuntimeError: Expected tensor for argument #2 'indices' to have scalar type Long; but got torch.IntTensor instead (while checking arguments for embeddingrenorm)