diff --git a/tests/test_models.py b/tests/test_models.py index 212c3e8f7c..b5b6445ad0 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -37,7 +37,7 @@ def get_exp(): def assert_ok_acc(trainer): # this model should get 0.80+ acc - assert trainer.tng_tqdm_dic['val_acc'] > 0.80 + assert trainer.tng_tqdm_dic['val_acc'] > 0.80, "model failed to get expected 0.80 validation accuracy" def test_cpu_model(): @@ -55,11 +55,8 @@ def test_cpu_model(): ) result = trainer.fit(model) + assert result == 1, 'cpu model failed to complete' - metrics = trainer.tng_tqdm_dic - print(metrics) - - assert result == 1 assert_ok_acc(trainer) @@ -84,7 +81,7 @@ def test_single_gpu_model(): result = trainer.fit(model) - assert result == 1 + assert result == 1, 'single gpu model failed to complete' assert_ok_acc(trainer) @@ -112,7 +109,7 @@ def test_multi_gpu_model_dp(): result = trainer.fit(model) - assert result == 1 + assert result == 1, 'multi-gpu dp model failed to complete' assert_ok_acc(trainer) @@ -141,7 +138,7 @@ def test_multi_gpu_model_ddp(): result = trainer.fit(model) - assert result == 1 + assert result == 1, 'multi-gpu ddp model failed to complete' assert_ok_acc(trainer) @@ -171,7 +168,7 @@ def test_amp_gpu_ddp(): result = trainer.fit(model) - assert result == 1 + assert result == 1, 'amp + ddp model failed to complete' assert_ok_acc(trainer) @@ -201,7 +198,7 @@ def test_amp_gpu_dp(): result = trainer.fit(model) - assert result == 1 + assert result == 1, 'amp + gpu model failed to complete' assert_ok_acc(trainer)