Right. Shuffling is definitely bad here. I completely forgot about that.
I had seen that example earlier. I was trying to use LSTM instead of LSTMCell. The batch_first thing is still confusing me a bit. Am I doing it correctly? By calling .unsqueeze(dim=0) when passing the input data, and calling .squeeze(dim=0) on the output.
Is there a better doc which I can read for understanding when to use retain_graph? Or could you please explain it in brief when to use it?
Best regards,
Jeet