missing eval mode in recurrent neural network

2025-07-07 01:54:41 +08:00 · 2018-06-03 09:22:20 +08:00
parent 7aac81272a
commit e217e42e60
1 changed files with 1 additions and 0 deletions
--- 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