fix export bug

2021-02-19 16:29:54 +08:00 · 2021-02-19 16:29:54 +08:00 · 0507402a0f
parent 33f302e5ff
commit 0507402a0f
1 changed files with 3 additions and 3 deletions
--- 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(