fix rare export error

2021-02-08 14:02:31 +08:00 · 2021-02-08 14:02:31 +08:00 · 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(