Open XXXxiaoxiaoxia opened 6 years ago
logpt=torch.gather(logpt,1,torch.LongTensor(target)) RuntimeError: Invalid index in gather at c:\users\administrator\downloads\new-builder\win-wheel\pytorch\aten\src\th\generic/THTensorMath.c:600
Process finished with exit code 1
check the training data to see the label of the data, do you run changeindex.py?
thank you,I have solved it.
I have another question, how to get train_new? run changeindex.py just can get val_new
logpt=torch.gather(logpt,1,torch.LongTensor(target)) RuntimeError: Invalid index in gather at c:\users\administrator\downloads\new-builder\win-wheel\pytorch\aten\src\th\generic/THTensorMath.c:600
Process finished with exit code 1