Merge pull request #2029 from WenmuZhou/fix_doc

Fix export bug
2021-02-19 16:34:23 +08:00 · 2021-02-19 16:34:23 +08:00 · 253cd0f770
parent 85d340bd27 0507402a0f
commit 253cd0f770
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(