"magic_pdf/git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "4c412b287883a2f263dccae917629c9ef7f49cf0"
Commit 9ce37af9 authored by thomwolf's avatar thomwolf
Browse files

oups

parent a40955f0
...@@ -895,7 +895,7 @@ def main(): ...@@ -895,7 +895,7 @@ def main():
tokenizer = BertTokenizer.from_pretrained(args.bert_model, do_lower_case=args.do_lower_case) tokenizer = BertTokenizer.from_pretrained(args.bert_model, do_lower_case=args.do_lower_case)
# Prepare model # 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))) # cache_dir=os.path.join(str(PYTORCH_PRETRAINED_BERT_CACHE), 'distributed_{}'.format(args.local_rank)))
if args.fp16: if args.fp16:
......
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