From a40955f0710478b141a7bbbca7edea45701eb833 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Tue, 18 Jun 2019 11:46:14 +0200 Subject: [PATCH] no need to duplicate models anymore --- examples/run_squad.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/run_squad.py b/examples/run_squad.py index a3525b1ee5..c3ae8c23b8 100644 --- a/examples/run_squad.py +++ b/examples/run_squad.py @@ -895,8 +895,8 @@ 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, - cache_dir=os.path.join(str(PYTORCH_PRETRAINED_BERT_CACHE), 'distributed_{}'.format(args.local_rank))) + 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: model.half()