removed dummy d

2019-07-24 10:59:15 -04:00 · 2019-07-24 10:59:15 -04:00 · 8e131f9d79
parent 88f064d276
commit 8e131f9d79
1 changed files with 5 additions and 1 deletions
--- 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()