Skip to content

GitLab

Commit a40955f0 authored by thomwolf's avatar thomwolf
Browse files

no need to duplicate models anymore

parent 3763f894
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/run_squad.py
View file @ a40955f0
...@@ -895,8 +895,8 @@ def main(): ...@@ -895,8 +895,8 @@ 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:
model.half() model.half()
......
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