MNIST Example: Incorrect Testing Accuracy

See issue for details; need to reset ClassErrorMeter.
edouardoyallon · Dec 29, 2018 · 57e6fae · 57e6fae
1 parent 15d37d4
commit 57e6fae
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/mnist.py b/examples/mnist.py
@@ -94,6 +94,7 @@ def on_end_epoch(state):
     def on_end(state):
         print('Training' if state['train'] else 'Testing', 'accuracy')
         print(classerr.value())
+        classerr.reset()
 
     optimizer = torch.optim.SGD(list(params.values()), lr=0.01, momentum=0.9,
                                 weight_decay=0.0005)