From 8e131f9d79a86908d7338a5d8dbbba6779a23f16 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 10:59:15 -0400 Subject: [PATCH] removed dummy d --- tests/debug.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/debug.py b/tests/debug.py index a7dd56aa14..6b27fd33c7 100644 --- a/tests/debug.py +++ b/tests/debug.py @@ -70,7 +70,11 @@ def main(): x, y = batch x = x.view(x.size(0), -1) out = model(x) - print(out) + + labels_hat = torch.argmax(out, dim=1) + val_acc = torch.sum(y == labels_hat).item() / (len(y) * 1.0) + val_acc = torch.tensor(val_acc) + print(val_acc) clear_tt_dir()