From d3651ba15ce3afaf83c0c43b68f1555b7546ace7 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 11:55:22 -0400 Subject: [PATCH] added test for model loading and predicting --- tests/debug.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/debug.py b/tests/debug.py index d8020f29fe..81611acc2f 100644 --- a/tests/debug.py +++ b/tests/debug.py @@ -86,7 +86,7 @@ def run_prediction(dataloader, trained_model): val_acc = torch.tensor(val_acc) val_acc = val_acc.item() - assert val_acc > 0.60, 'this model is expected to get > 0.7 in test set' + assert val_acc > 0.60, f'this model is expected to get > 0.7 in test set (it got {val_acc})' def main():