-
Notifications
You must be signed in to change notification settings - Fork 8
Description
i have data customer same form but when train, code error :
File "train.py", line 95, in main hit, mrr = train_test(model, train_data, test_data) File "/mnt/nvme0n1/phuongnam/SBR/model.py", line 185, in train_test targets, scores = forward(model, i, train_data) File "/mnt/nvme0n1/phuongnam/SBR/model.py", line 167, in forward hidden = model(items, A) File "/mnt/nvme0n1/miniconda3/envs/namtp/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/mnt/nvme0n1/phuongnam/SBR/model.py", line 130, in forward hidden = self.embedding(inputs) File "/mnt/nvme0n1/miniconda3/envs/namtp/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/mnt/nvme0n1/miniconda3/envs/namtp/lib/python3.8/site-packages/torch/nn/modules/sparse.py", line 156, in forward return F.embedding( File "/mnt/nvme0n1/miniconda3/envs/namtp/lib/python3.8/site-packages/torch/nn/functional.py", line 1916, in embedding return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse) IndexError: index out of range in self
How can I choose n_node correctly?