Generalize problem_type to all sequence classification models (#14180)

* Generalize problem_type to all classification models * Missing import * Deberta BC and fix tests * Fix template * Missing imports * Revert change to reformer test * Fix style
2021-10-29 10:32:56 -04:00
parent 4ab6a4a086
commit c28bc80bbb
38 changed files with 474 additions and 191 deletions
--- a/tests/test_modeling_squeezebert.py
+++ b/tests/test_modeling_squeezebert.py
@@ -232,7 +232,6 @@ class SqueezeBertModelTest(ModelTesterMixin, unittest.TestCase):
    test_torchscript = True
    test_resize_embeddings = True
    test_head_masking = False
-    test_sequence_classification_problem_types = True

    def setUp(self):
        self.model_tester = SqueezeBertModelTester(self)