missing eval mode in recurrent neural network

arisliang · arisliang · commit e217e42e600e · 2018-06-03T09:22:20.000+08:00
diff --git a/tutorials/02-intermediate/recurrent_neural_network/main.py b/tutorials/02-intermediate/recurrent_neural_network/main.py
@@ -85,6 +85,7 @@ def forward(self, x):
                    .format(epoch+1, num_epochs, i+1, total_step, loss.item()))
 
 # Test the model
+model.eval()
 with torch.no_grad():
     correct = 0
     total = 0
