tiny fix (#3024)

4a11ec10 · tianxin · Yibing Liu · 7fdde01a · 4a11ec10
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

PaddleNLP/language_representations_kit/BERT/reader/pretraining.py ...P/language_representations_kit/BERT/reader/pretraining.py +1 -1

未找到文件。
--- a/PaddleNLP/language_representations_kit/BERT/reader/pretraining.py
+++ b/PaddleNLP/language_representations_kit/BERT/reader/pretraining.py
@@ -159,7 +159,7 @@ class DataReader(object):

            src_ids = origin_src_ids[:origin_sep_index + 1] + pair_src_ids[
                pair_sep_index + 1:]
-            if len(src_ids) >= self.max_seq_len:
+            if len(src_ids) > self.max_seq_len:
                miss_num += 1
                continue
            sent_ids = [0] * len(origin_src_ids[:origin_sep_index + 1]) + [