diff --git a/tests/unit/algo/detection/test_dfine.py b/tests/unit/algo/detection/test_dfine.py index 91b26dd01c..6ba7dc838b 100644 --- a/tests/unit/algo/detection/test_dfine.py +++ b/tests/unit/algo/detection/test_dfine.py @@ -21,9 +21,6 @@ class TestDFine: @pytest.mark.parametrize( "model", [ - DFine(label_info=3, model_name="dfine_hgnetv2_n"), - DFine(label_info=3, model_name="dfine_hgnetv2_s"), - DFine(label_info=3, model_name="dfine_hgnetv2_m"), DFine(label_info=3, model_name="dfine_hgnetv2_l"), DFine(label_info=3, model_name="dfine_hgnetv2_x"), ], @@ -36,9 +33,6 @@ def test_loss(self, model, fxt_data_module): @pytest.mark.parametrize( "model", [ - DFine(label_info=3, model_name="dfine_hgnetv2_n"), - DFine(label_info=3, model_name="dfine_hgnetv2_s"), - DFine(label_info=3, model_name="dfine_hgnetv2_m"), DFine(label_info=3, model_name="dfine_hgnetv2_l"), DFine(label_info=3, model_name="dfine_hgnetv2_x"), ], @@ -53,9 +47,6 @@ def test_predict(self, model, fxt_data_module): @pytest.mark.parametrize( "model", [ - DFine(label_info=3, model_name="dfine_hgnetv2_n"), - DFine(label_info=3, model_name="dfine_hgnetv2_s"), - DFine(label_info=3, model_name="dfine_hgnetv2_m"), DFine(label_info=3, model_name="dfine_hgnetv2_l"), DFine(label_info=3, model_name="dfine_hgnetv2_x"), ],