Skip to content

GitLab

Unverified Commit e5fe2bb5 authored by Thomas Wolf's avatar Thomas Wolf Committed by GitHub
Browse files

Merge pull request #745 from leimao/leimao 

fix evaluation bug
parents dad3c7a4 64b2a828
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
examples/run_squad.py
View file @ e5fe2bb5
...@@ -336,7 +336,9 @@ def main(): ...@@ -336,7 +336,9 @@ def main():
output_args_file = os.path.join(args.output_dir, 'training_args.bin') output_args_file = os.path.join(args.output_dir, 'training_args.bin')
torch.save(args, output_args_file) torch.save(args, output_args_file)
else: 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) model.to(device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment