fix test examples et model pretrained

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

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()