Hi, thank you very much for taking a look at this.
The shape of X_train at y_pred = lstm_model(X_train) is:
torch.Size([1, 1969875, 65])

Then y_pred is of shape:
torch.Size([1969875])

The batch_size I passed only as parameter when creating my model, I assume that the model take by itself the “batch sizes” of X_train?
Thanks a lot for your help.