no need to duplicate models anymore

a40955f0 · thomwolf · 3763f894 · a40955f0
Commit a40955f0 authored Jun 18, 2019 by thomwolf
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

examples/run_squad.py examples/run_squad.py +2 -2

No files found.
--- 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()