Fix pir dy2st train (#13853)

2026-10-29 18:48:40 +08:00 · 2024-09-11 18:54:37 +08:00
parent 33d5570249
commit 4832bb62ad
1 changed files with 0 additions and 2 deletions
@@ -372,8 +372,6 @@ class SLAHead(nn.Layer):
            )
            pre_chars = paddle.zeros(shape=[batch_size], dtype="int32")
            max_text_length = paddle.to_tensor(self.max_text_length)
-            # for export
-            loc_step, structure_step = None, None
            for i in range(max_text_length + 1):
                hidden, structure_step, loc_step = self._decode(pre_chars, fea, hidden)
                pre_chars = structure_step.argmax(axis=1, dtype="int32")