diff --git a/tutorials/02-intermediate/recurrent_neural_network/main.py b/tutorials/02-intermediate/recurrent_neural_network/main.py index 9b8685c..c138c5a 100644 --- a/tutorials/02-intermediate/recurrent_neural_network/main.py +++ b/tutorials/02-intermediate/recurrent_neural_network/main.py @@ -85,6 +85,7 @@ for epoch in range(num_epochs): .format(epoch+1, num_epochs, i+1, total_step, loss.item())) # Test the model +model.eval() with torch.no_grad(): correct = 0 total = 0