From 98437b401f077301a22847ad3a85d7775ddb9ce9 Mon Sep 17 00:00:00 2001
From: san999p <57917118+san999p@users.noreply.github.com>
Date: Tue, 26 May 2020 19:14:24 -0500
Subject: [PATCH] return accuracy instead of correct examples

---
 mnist/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mnist/main.py b/mnist/main.py
index fd2f5a689e..305b718a5d 100644
--- a/mnist/main.py
+++ b/mnist/main.py
@@ -66,7 +66,7 @@ def test(args, model, device, test_loader):
     print('\nTest set: Average loss: {:.4f}, Accuracy: {}/{} ({:.0f}%)\n'.format(
         test_loss, correct, len(test_loader.dataset),
         100. * correct / len(test_loader.dataset)))
-    return test_loss, correct
+    return test_loss, round(100. * correct / len(test_loader.dataset),2)
 
 
 def main():