fixing PYTORCH_PRETRAINED_BERT_CACHE use in examples

2019-03-06 10:21:24 +01:00
parent 2dd8f524f5
commit 994d86609b
3 changed files with 3 additions and 3 deletions
--- a/examples/run_swag.py
+++ b/examples/run_swag.py
@@ -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()