Hey,
I’ve been trying to transfer my LSTM model to CUDA, but it messes with the expected LSTM hidden dimensions. When on CPU the expected input (layers, batch, hidden: 4x2x8) just works, but when enable cuda (using model.cuda()) it expects a 1x2x8 dimension. Why?
RuntimeError: Expected hidden size (1, 2, 8), got (4, 2, 8)

Thanks in advance,
David