From 788e632622b27f6665e8e85ae23f3f93552a1dd7 Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Fri, 11 Oct 2019 18:04:29 -0400 Subject: [PATCH] [ner] Honor args.overwrite_cache --- examples/run_ner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/run_ner.py b/examples/run_ner.py index 0e40ad02a6..fdf2f1924a 100644 --- a/examples/run_ner.py +++ b/examples/run_ner.py @@ -252,7 +252,7 @@ def load_and_cache_examples(args, tokenizer, labels, pad_token_label_id, mode): cached_features_file = os.path.join(args.data_dir, "cached_{}_{}_{}".format(mode, list(filter(None, args.model_name_or_path.split("/"))).pop(), str(args.max_seq_length))) - if os.path.exists(cached_features_file): + if os.path.exists(cached_features_file) and not args.overwrite_cache: logger.info("Loading features from cached file %s", cached_features_file) features = torch.load(cached_features_file) else: