From 3d5f2913864de28a57a339c4c0c9f7b6000a7d03 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Fri, 5 Jul 2019 17:22:15 +0200 Subject: [PATCH] updates to run_glue --- examples/run_glue.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/run_glue.py b/examples/run_glue.py index 1c2e921ef7..8dd845a553 100644 --- a/examples/run_glue.py +++ b/examples/run_glue.py @@ -213,11 +213,12 @@ def load_and_cache_examples(args, task, tokenizer, eval=False): str(task))) if os.path.exists(cached_features_file): + logger.info("Loading features from cached file %s", cached_features_file) features = torch.load(cached_features_file) else: features = convert_examples_to_features(examples, label_list, args.max_seq_length, tokenizer, output_mode) if args.local_rank == -1 or torch.distributed.get_rank() == 0: - logger.info(" Saving eval features into cached file %s", cached_features_file) + logger.info("Saving features into cached file %s", cached_features_file) torch.save(features, cached_features_file) return features