fix test examples et model pretrained

This commit is contained in:
thomwolf committed 2019-07-11 22:29:08 +02:00
1 parent 92a782b108
commit c6bf1a400d
3 files changed
+4 -2

No files matched your search

+2 -1
View File
@@ -56,7 +56,8 @@ class ExamplesTests(unittest.TestCase):
"--learning_rate=1e-4",
"--max_steps=10",
"--warmup_steps=2",
"--overwrite_output_dir"]
"--overwrite_output_dir",
"--seed=42"]
model_name = "--model_name=bert-base-uncased"
with patch.object(sys, 'argv', testargs + [model_name]):
result = run_glue.main()