From 97d9299356596610be86ac84edb425853cece0cd Mon Sep 17 00:00:00 2001
From: kinghuin <kinghuin_chull@163.com>
Date: Tue, 31 Dec 2019 10:16:49 +0800
Subject: [PATCH] fix seq predict bug (#294)

---
 paddlehub/finetune/task/sequence_task.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddlehub/finetune/task/sequence_task.py b/paddlehub/finetune/task/sequence_task.py
index c524468f..05d0393c 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
-- 
GitLab