fix evaluation bug

2019-07-01 14:56:24 -07:00
parent dad3c7a485
commit 64b2a828c0
1 changed files with 3 additions and 1 deletions
--- 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)