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

using multi_gpu consistently (#8446)

Browse Source 
* s|multiple_gpu|multi_gpu|g; s|multigpu|multi_gpu|g'

* doc
This commit is contained in:
Stas Bekman
2020-11-10 10:23:58 -08:00
committed by GitHub
parent b93569457f
commit 02bdfc0251
22 changed files with 117 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_trainer_distributed.py
View File

@@ -2,7 +2,7 @@ import sys
from typing import Dict
from transformers import EvalPrediction, HfArgumentParser, TrainingArguments, is_torch_available
from transformers.testing_utils import TestCasePlus, execute_subprocess_async, require_torch_multigpu
from transformers.testing_utils import TestCasePlus, execute_subprocess_async, require_torch_multi_gpu
from transformers.utils import logging
@@ -44,7 +44,7 @@ if is_torch_available():
class TestTrainerDistributed(TestCasePlus):
@require_torch_multigpu
@require_torch_multi_gpu
def test_trainer(self):
distributed_args = f"""