Pata_Naheen:

u change something o


I only changed
X = torch.FloatTensor(X)
y = torch.LongTensor(Y)
to a some real tensors:
X = torch.randn(10, 1, 6)
y = torch.randn(10, 1, 2)
And everything works.
I changed 1000 to 10 just to make it end sooner.