Update pages/19_RNN_LSTM_Shakespeare.py

pages/19_RNN_LSTM_Shakespeare.py

@@ -87,15 +87,14 @@ if st.button("Train and Generate"):
 
             # Training the model
             for epoch in range(num_epochs):
-                h = (torch.zeros(num_layers, X_tensor.size(0), hidden_size), torch.zeros(num_layers, X_tensor.size(0), hidden_size))
+                h = (torch.zeros(num_layers, 1, hidden_size), torch.zeros(num_layers, 1, hidden_size))
                 epoch_loss = 0
                 for i in range(len(dataX)):
                     inputs = X_tensor[i].unsqueeze(0)
                     targets = Y_tensor[i].unsqueeze(0)
 
                     # Forward pass
-                    outputs, h = model(inputs, h)
-                    h = (h[0].detach(), h[1].detach())
+                    outputs, h = model(inputs, (h[0].detach(), h[1].detach()))
                     loss = criterion(outputs, targets)
 
                     # Backward pass and optimization