fixing PYTORCH_PRETRAINED_BERT_CACHE use in examples
This commit is contained in:
1 parent
2dd8f524f5
commit
994d86609b
3 files changed
+3
-3
No files matched your search
@@ -367,7 +367,7 @@ def main():
|
||||
|
||||
# Prepare model
|
||||
model = BertForMultipleChoice.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)),
|
||||
num_choices=4)
|
||||
if args.fp16:
|
||||
model.half()
|
||||
|
||||
Reference in new issue
Block a user