Merge pull request #745 from leimao/leimao

fix evaluation bug

Merge pull request #745 from leimao/leimao
fix evaluation bug
e5fe2bb5 · Thomas Wolf · GitHub · dad3c7a4 · 64b2a828 · e5fe2bb5
Unverified Commit e5fe2bb5 authored Jul 05, 2019 by Thomas Wolf Committed by GitHub Jul 05, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

examples/run_squad.py examples/run_squad.py +3 -1

No files found.
--- a/examples/run_squad.py
+++ b/examples/run_squad.py
@@ -336,7 +336,9 @@ def main():
        output_args_file = os.path.join(args.output_dir, 'training_args.bin')
        torch.save(args, output_args_file)
    else:
-        model = BertForQuestionAnswering.from_pretrained(args.bert_model)
+        # Load a trained model and vocabulary that you have fine-tuned
+        model = BertForQuestionAnswering.from_pretrained(args.output_dir)
+        tokenizer = BertTokenizer.from_pretrained(args.output_dir, do_lower_case=args.do_lower_case)

    model.to(device)