skip cuda

Signed-off-by: lizz <lizz@sensetime.com>
2021-04-05 21:25:16 +08:00 · 2021-04-05 21:25:16 +08:00 · 7e281521cf
parent 06fee8328b
commit 7e281521cf
1 changed files with 1 additions and 0 deletions
--- a/tests/test_models/test_detector.py
+++ b/tests/test_models/test_detector.py
@ -261,6 +261,7 @@ def test_psenet(cfg_file):
    detector.show_result(img, results)


+@pytest.mark.skipif(not torch.cuda.is_available(), reason='requires cuda')
@pytest.mark.parametrize('cfg_file', [
    'textdet/dbnet/dbnet_r18_fpnc_1200e_icdar2015.py',
    'textdet/dbnet/dbnet_r50dcnv2_fpnc_1200e_icdar2015.py'