From 64b2a828c0b16368810d15c6fff1640e33a34125 Mon Sep 17 00:00:00 2001 From: Lei Mao Date: Mon, 1 Jul 2019 14:56:24 -0700 Subject: [PATCH] fix evaluation bug --- examples/run_squad.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/run_squad.py b/examples/run_squad.py index bf1763e884..fb8d08947c 100644 --- 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)