diff --git a/examples/run_squad.py b/examples/run_squad.py index c3ae8c23..d4bfc025 100644 --- a/examples/run_squad.py +++ b/examples/run_squad.py @@ -895,7 +895,7 @@ def main(): tokenizer = BertTokenizer.from_pretrained(args.bert_model, do_lower_case=args.do_lower_case) # Prepare model - model = BertForQuestionAnswering.from_pretrained(args.bert_model), + model = BertForQuestionAnswering.from_pretrained(args.bert_model) # cache_dir=os.path.join(str(PYTORCH_PRETRAINED_BERT_CACHE), 'distributed_{}'.format(args.local_rank))) if args.fp16: