[ner] Honor args.overwrite_cache

This commit is contained in:
Julien Chaumond authored and thomwolf committed 2019-10-15 09:17:31 +02:00
1 parent 0f9ebb0b43
commit 788e632622
1 file changed
+1 -1
+1 -1
View File
@@ -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: