fix

train.py

@@ -105,7 +105,7 @@ for epoch in range(num_epochs):
         # Reset hidden states if needed
         h_states = h_states if (step % reset_state_every!= 0) else None
         str_states = (
-            ''.join(['{:.3f}, '.format(h_states[0][0][0][j].item()) for j in range(10)] if h_states is not None else 'None'
+            ''.join(['{:.3f}, '.format(h_states[0][0][0][j].item()) for j in range(10)] if h_states is not None else 'None')
         )
 
         optimizer.zero_grad()