added test for model loading and predicting

2019-07-24 11:56:16 -04:00 · 2019-07-24 11:56:16 -04:00 · 8781d8aeab
parent d3651ba15c
commit 8781d8aeab
1 changed files with 2 additions and 0 deletions
--- 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})'