fix rare export error

4 years ago · 9cc6363b0e
parent 9050c6f933
commit 9cc6363b0e
1 changed files with 3 additions and 0 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(