From 58359ca803cce2a6d001fa1622e18d15af75d7c7 Mon Sep 17 00:00:00 2001 From: manfeibaigithub Date: Wed, 24 Apr 2024 23:28:38 +0000 Subject: [PATCH] down into cpp --- test/test_test_mnist.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_test_mnist.py b/test/test_test_mnist.py index 7bc355ae7ba..c3613045682 100644 --- a/test/test_test_mnist.py +++ b/test/test_test_mnist.py @@ -91,8 +91,8 @@ def test(): test_loss /= 20 test_losses.append(test_loss) print('\nTest set: Avg. loss: {:.4f}, Accuracy: {}/{} ({:.0f}%)\n'.format( - test_loss, correct, len(test_loader.dataset), - 100. * correct / len(test_loader.dataset))) + test_loss, correct, 20, + 100. * correct / 20)) # run test model test()