From ca1835e063d2c6cc1a5065e0a830bbedc8493687 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 14:14:36 -0400 Subject: [PATCH] fixed amp bug --- tests/test_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_models.py b/tests/test_models.py index f9b7b91c..fa478b93 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -106,7 +106,7 @@ def test_amp_gpu_dp(): use_amp=True ) except Exception as e: - assert 'https://github.com/NVIDIA/apex/issues/227' in e + assert 'https://github.com/NVIDIA/apex/issues/227' in str(e) run_gpu_model_test(trainer_options)