MGT.py里的代码报错

[shuqincao]

MGT.py文件里的第87,88行“inputs_extras_embedding = torch.cat([self.embedding_modulesi
for i in range(len(self.num_embeddings))] + [inputs_pe], dim=-1)”
代码报错，报错详情如下：

---------- Training ----------
num_samples: 1188, num_batches: 594
0%| | 0/594 [00:00<?, ?it/s]Traceback (most recent call last):
File "F:/orginalCode/MGT-main/main.py", line 236, in
train(args, logger)
File "F:/orginalCode/MGT-main/main.py", line 156, in train
criterion, optimizer, scheduler, args)
File "F:/orginalCode/MGT-main/main.py", line 91, in train_epoch
outputs = model(inputs, targets, *extras, **statics)
File "D:\programfiles\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "F:\orginalCode\MGT-main\models\MGT.py", line 503, in forward
z_inputs, z_targets = self.temporal_embedding(extras) # (B, P, d_model), (B, Q, d_model)
File "D:\programfiles\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "F:\orginalCode\MGT-main\models\MGT.py", line 88, in forward
for i in range(len(self.num_embeddings))] + [inputs_pe], dim=-1)
File "F:\orginalCode\MGT-main\models\MGT.py", line 88, in
for i in range(len(self.num_embeddings))] + [inputs_pe], dim=-1)
File "D:\programfiles\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "D:\programfiles\Anaconda3\lib\site-packages\torch\nn\modules\sparse.py", line 126, in forward
self.norm_type, self.scale_grad_by_freq, self.sparse)
File "D:\programfiles\Anaconda3\lib\site-packages\torch\nn\functional.py", line 1852, in embedding
return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)
TypeError: embedding(): argument 'indices' (position 2) must be Tensor, not list
0%| | 0/594 [00:13<?, ?it/s]

Process finished with exit code 1

[lonicera-yx]

我更新了requirements.txt，给出了各package的版本号，你参考一下，我在自己的机器上跑是没问题的。

[shuqincao]

这个代码我还是跑不起来，我可以加你的微信吗？我的1033898863