c-my/transformers
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixing PYTORCH_PRETRAINED_BERT_CACHE use in examples

Browse Source
This commit is contained in:
thomwolf committed 2019-03-06 10:21:24 +01:00
1 parent 2dd8f524f5
commit 994d86609b
3 files changed
+3 -3
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

examples/run_squad.py
+1 -1
View File
@@ -894,7 +894,7 @@ def main():
# Prepare model
model = BertForQuestionAnswering.from_pretrained(args.bert_model,
cache_dir=os.path.join(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:
model.half()