From 8781d8aeabbf53c97ef445b6cd4dad7591ec7fcb Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 24 Jul 2019 11:56:16 -0400
Subject: [PATCH] added test for model loading and predicting

---
 tests/debug.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/debug.py b/tests/debug.py
index 81611acc2f..978322dbab 100644
--- a/tests/debug.py
+++ b/tests/debug.py
@@ -86,6 +86,8 @@ def run_prediction(dataloader, trained_model):
     val_acc = torch.tensor(val_acc)
     val_acc = val_acc.item()
 
+    print(val_acc)
+
     assert val_acc > 0.60, f'this model is expected to get > 0.7 in test set (it got {val_acc})'