SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Reactivate Megatron tests an use less workers

Browse Source
This commit is contained in:
Sylvain Gugger
2021-04-09 18:09:53 -04:00
parent 716120cbd6
commit 26212c14e5
2 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.circleci/config.yml
View File

@@ -145,7 +145,7 @@ jobs:
key: v0.4-torch-{{ checksum "setup.py" }} key: v0.4-torch-{{ checksum "setup.py" }}
paths: paths:
- '~/.cache/pip' - '~/.cache/pip'
- run: python -m pytest -n 8 --dist=loadfile -s --make-reports=tests_torch ./tests/ | tee tests_output.txt - run: python -m pytest -n 4 --dist=loadfile -s --make-reports=tests_torch ./tests/ | tee tests_output.txt
- store_artifacts: - store_artifacts:
path: ~/transformers/tests_output.txt path: ~/transformers/tests_output.txt
- store_artifacts: - store_artifacts:

1
tests/test_modeling_megatron_bert.py
View File

@@ -265,7 +265,6 @@ class MegatronBertModelTester:
@require_torch @require_torch
@unittest.skip("Temporary skip to make the CI pass reliably.")
class MegatronBertModelTest(ModelTesterMixin, unittest.TestCase): class MegatronBertModelTest(ModelTesterMixin, unittest.TestCase):
all_model_classes = ( all_model_classes = (