This commit is contained in:
LysandreJik committed 2019-12-04 16:24:15 -05:00
1 parent a7ca6d738b
commit f7e4a7cdfa
6 files changed
+191 -24

No files matched your search

+1 -2
View File
@@ -72,8 +72,7 @@ class ExamplesTests(unittest.TestCase):
logger.addHandler(stream_handler)
testargs = ["run_squad.py",
"--train_file=./examples/tests_samples/SQUAD/dev-v2.0-small.json",
"--predict_file=./examples/tests_samples/SQUAD/dev-v2.0-small.json",
"--data_dir=./examples/tests_samples/SQUAD",
"--model_name=bert-base-uncased",
"--output_dir=./examples/tests_samples/temp_dir",
"--max_steps=10",