fix export bug

0507402a · WenmuZhou · 33f302e5 · 0507402a
隐藏空白更改
内联并排

Showing with 3 addition and 3 deletion

ppocr/modeling/heads/rec_att_head.py ppocr/modeling/heads/rec_att_head.py +3 -3

未找到文件。
--- a/ppocr/modeling/heads/rec_att_head.py
+++ b/ppocr/modeling/heads/rec_att_head.py
@@ -57,6 +57,9 @@ class AttentionHead(nn.Layer):
        else:
            targets = paddle.zeros(shape=[batch_size], dtype="int32")
            probs = None
+            char_onehots = None
+            outputs = None
+            alpha = None

            for i in range(num_steps):
                char_onehots = self._char_to_onehot(
@@ -146,9 +149,6 @@ class AttentionLSTM(nn.Layer):
        else:
            targets = paddle.zeros(shape=[batch_size], dtype="int32")
            probs = None
-            char_onehots = None
-            outputs = None
-            alpha = None

            for i in range(num_steps):
                char_onehots = self._char_to_onehot(