Reactivate Megatron tests an use less workers

2021-04-09 18:09:53 -04:00
parent 716120cbd6
commit 26212c14e5
2 changed files with 1 additions and 2 deletions
--- a/tests/test_modeling_megatron_bert.py
+++ b/tests/test_modeling_megatron_bert.py
@@ -265,7 +265,6 @@ class MegatronBertModelTester:


@require_torch
-@unittest.skip("Temporary skip to make the CI pass reliably.")
 class MegatronBertModelTest(ModelTesterMixin, unittest.TestCase):

    all_model_classes = (