We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7aac812 commit e217e42Copy full SHA for e217e42
tutorials/02-intermediate/recurrent_neural_network/main.py
@@ -85,6 +85,7 @@ def forward(self, x):
85
.format(epoch+1, num_epochs, i+1, total_step, loss.item()))
86
87
# Test the model
88
+model.eval()
89
with torch.no_grad():
90
correct = 0
91
total = 0
0 commit comments