Hi, I am trying to implement an LSTM model to predict coronavirus cases and I got this error : **RuntimeError: Expected hidden[0] size (2, 134, 14), got (2, 14, 14)**

Here is some informations :

Len test : 47

Len validation : 37

Len train : 149

seq_len = 14

input_size = 1

output_size = 1

hidden_dim = 14

n_layers = 2

dropout = 0.4

num_epochs = 10

and the model is that :

```
class LSTM(nn.Module):
def __init__(self, input_size, output_size, hidden_dim, n_layers,seq_len,dropout):
super(LSTM, self).__init__()
self.hidden_dim=hidden_dim
self.lstm = nn.LSTM(input_size, hidden_dim, n_layers, batch_first=True, dropout = dropout)
self.fc = nn.Linear(hidden_dim, output_size)
def reset_hidden_state(self):
self.hidden = (
torch.zeros(n_layers, seq_len, hidden_dim),
torch.zeros(n_layers, seq_len, hidden_dim))
def forward(self, sequences):
lstm_out, self.hidden = self.lstm(
sequences.view(len(sequences), seq_len, -1),
self.hidden)
last_time_step = lstm_out.view(seq_len, len(sequences), self.n_hidden)[-1]
y_pred = self.linear(last_time_step)
return y_pred
```

What can I do to solve this problem ?