Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
2819da02
"vscode:/vscode.git/clone" did not exist on "9678c49419ca81b854d80f77be2a9c027f9b5513"
Unverified
Commit
2819da02
authored
Nov 18, 2020
by
Stas Bekman
Committed by
GitHub
Nov 18, 2020
Browse files
[s2s] broken test (#8613)
parent
9fa3ed1a
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
1 deletion
+8
-1
examples/seq2seq/test_finetune_trainer.py
examples/seq2seq/test_finetune_trainer.py
+8
-1
No files found.
examples/seq2seq/test_finetune_trainer.py
View file @
2819da02
...
...
@@ -4,7 +4,13 @@ from unittest.mock import patch
from
transformers
import
BertTokenizer
,
EncoderDecoderModel
from
transformers.file_utils
import
is_datasets_available
from
transformers.testing_utils
import
TestCasePlus
,
execute_subprocess_async
,
get_gpu_count
,
slow
from
transformers.testing_utils
import
(
TestCasePlus
,
execute_subprocess_async
,
get_gpu_count
,
require_torch_non_multi_gpu_but_fix_me
,
slow
,
)
from
transformers.trainer_callback
import
TrainerState
from
transformers.trainer_utils
import
set_seed
...
...
@@ -46,6 +52,7 @@ class TestFinetuneTrainer(TestCasePlus):
assert
"test_results.json"
in
contents
@
slow
@
require_torch_non_multi_gpu_but_fix_me
def
test_finetune_bert2bert
(
self
):
if
not
is_datasets_available
():
return
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment