From c45d0cf60f6b62ad5872e41ef79a0ad039994cd3 Mon Sep 17 00:00:00 2001 From: Simone Primarosa Date: Sun, 29 Dec 2019 13:45:45 +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 51701ded73..d16d72f6cb 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/%d" % (ex_index, len(examples))) + logger.info("Writing example %d/%d" % (ex_index, len(self.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)