From 9c1786d070afa595862469ccab5ff3a709d72ab7 Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Sat, 23 Jan 2021 06:36:22 -0600
Subject: [PATCH] Measure accuracy in percentages
---
predtuner/torchutil/common_qos.py | 2 +-
test/test_torchapp.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/predtuner/torchutil/common_qos.py b/predtuner/torchutil/common_qos.py
index f7c9ccd..a6a19f6 100644
--- a/predtuner/torchutil/common_qos.py
+++ b/predtuner/torchutil/common_qos.py
@@ -4,4 +4,4 @@ from torch import Tensor
def accuracy(output: Tensor, target: Tensor) -> float:
_, pred_labels = output.max(1)
n_correct = (pred_labels == target).sum().item()
- return n_correct / len(output)
+ return n_correct / len(output) * 100
diff --git a/test/test_torchapp.py b/test/test_torchapp.py
index 0184093..1621a08 100644
--- a/test/test_torchapp.py
+++ b/test/test_torchapp.py
@@ -44,7 +44,7 @@ class TestTorchApp(unittest.TestCase):
def test_baseline_qos(self):
app = self.get_app()
qos, _ = app.measure_qos_perf({}, False)
- self.assertAlmostEqual(qos, 0.88)
+ self.assertAlmostEqual(qos, 88.0)
def test_tuning(self):
app = TorchApp(
--
GitLab