search

nnzhan / MTGNN

MIT License
798 stars 222 forks source link

Program Running Problem #17

Closed zhp66 closed 3 years ago

zhp66 commented 3 years ago

Hello, I configured the environment according to the configuration required by the program, but the program still can't run begin training Traceback (most recent call last): File "train_single_step.py", line 218, in val_acc, val_rae, val_corr, test_acc, test_rae, test_corr = main() File "train_single_step.py", line 178, in main train_loss = train(Data, Data.train[0], Data.train[1], model, criterion, optim, args.batch_size) File "train_single_step.py", line 77, in train tx = X[:, :, id, :] IndexError: tensors used as indices must be long, byte or bool tensors

kaisazhang commented 3 years ago

Hello, how did you solve this problem？

zixuanxuanxuanxuan commented 1 year ago

我也遇到了同样的问题