diff --git a/paddlehub/finetune/task/sequence_task.py b/paddlehub/finetune/task/sequence_task.py index c524468f63721efeb31fc90771a7bf4dca53dc34..05d0393c459810267e89f9cc44b7a43d4a35b239 100644 --- a/paddlehub/finetune/task/sequence_task.py +++ b/paddlehub/finetune/task/sequence_task.py @@ -232,6 +232,6 @@ class SequenceLabelTask(BaseTask): for length in seq_lens: seq_infers = batch_infers[current_id:current_id + length] seq_result = list(map(id2label.get, seq_infers[1:-1])) - current_id += self.max_seq_len + current_id += length if self.add_crf else self.max_seq_len results.append(seq_result) return results