Update pages/19_RNN_LSTM_Shakespeare.py

pages/19_RNN_LSTM_Shakespeare.py

@@ -90,8 +90,8 @@ if st.button("Train and Generate"):
                 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)
+                    inputs = X_tensor[i].unsqueeze(0)  # Shape: (1, seq_length, 1)
+                    targets = Y_tensor[i].unsqueeze(0)  # Shape: (1,)
 
                     # Forward pass
                     outputs, h = model(inputs, (h[0].detach(), h[1].detach()))