oups
This commit is contained in:
parent
a40955f071
commit
9ce37af99b
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue