riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

[T5, examples] replace heavy t5 models with tiny random models (#3556)

Browse Source 
* replace heavy t5 models with tiny random models as was done by sshleifer

* fix isort
This commit is contained in:
Patrick von Platen 2020-04-02 12:34:05 +02:00 committed by GitHub
parent a4ee4da18a
commit ab5d06a094
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
examples/summarization/t5/test_t5_examples.py
View File

@ -29,7 +29,14 @@ class TestT5Examples(unittest.TestCase):
output_file_name = Path(tempfile.gettempdir()) / "utest_output_t5_sum.hypo"
score_file_name = Path(tempfile.gettempdir()) / "utest_score_t5_sum.hypo"
testargs = ["evaluate_cnn.py", "t5-small", str(tmp), str(output_file_name), str(tmp), str(score_file_name)]
testargs = [
"evaluate_cnn.py",
"patrickvonplaten/t5-tiny-random",
str(tmp),
str(output_file_name),
str(tmp),
str(score_file_name),
]
with patch.object(sys, "argv", testargs):
run_generate()

2
examples/translation/t5/test_t5_examples.py
View File

@ -37,7 +37,7 @@ class TestT5Examples(unittest.TestCase):
testargs = [
"evaluate_wmt.py",
"t5-small",
"patrickvonplaten/t5-tiny-random",
str(tmp_source),
str(output_file_name),
str(tmp_target),