From bf89be77b9a5917b233227e61d4812d7d2ac5567 Mon Sep 17 00:00:00 2001 From: Simone Primarosa Date: Sun, 29 Dec 2019 13:41:46 +0100 Subject: [PATCH] Improve logging message in the single sentence classification processor --- src/transformers/data/processors/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/data/processors/utils.py b/src/transformers/data/processors/utils.py index a7244ab39c..51701ded73 100644 --- a/src/transformers/data/processors/utils.py +++ b/src/transformers/data/processors/utils.py @@ -253,7 +253,7 @@ class SingleSentenceClassificationProcessor(DataProcessor): features = [] for (ex_index, (input_ids, example)) in enumerate(zip(all_input_ids, self.examples)): if ex_index % 10000 == 0: - logger.info("Writing example %d", ex_index) + logger.info("Writing example %d/%d" % (ex_index, len(examples))) # The mask has 1 for real tokens and 0 for padding tokens. Only real # tokens are attended to. attention_mask = [1 if mask_padding_with_zero else 0] * len(input_ids)