diff --git a/tests/test_models.py b/tests/test_models.py index fa478b93..be9db386 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -281,7 +281,7 @@ def run_prediction(dataloader, trained_model): print(val_acc) - assert val_acc > 0.70, f'this model is expected to get > 0.7 in test set (it got {val_acc})' + assert val_acc > 0.60, f'this model is expected to get > 0.6 in test set (it got {val_acc})' def assert_ok_acc(trainer):