From e217e42e600ece714d047a6c7bcb098728cfa3ae Mon Sep 17 00:00:00 2001 From: Liang Yi Date: Sun, 3 Jun 2018 09:22:20 +0800 Subject: [PATCH] missing eval mode in recurrent neural network --- tutorials/02-intermediate/recurrent_neural_network/main.py | 1 + 1 file changed, 1 insertion(+) 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