Hello Kabilan!



 titoniubo:

train_y = torch.LongTensor(Y_train).long().requires_grad()





 kabilan:

Remove long function


No, this isn’t correct.  First, calling the .long() method on a
LongTensor is a no-op (does nothing).  Second, the train_y
are to be used as categorical class labels, so they have to be
a LongTensor.
Best.
K. Frank