From ed2b527c9ff185425b7ea42a6f88926294899e52 Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Mon, 8 Feb 2021 14:06:10 +0800 Subject: [PATCH] fix rare export error --- ppocr/modeling/heads/rec_att_head.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/ppocr/modeling/heads/rec_att_head.py b/ppocr/modeling/heads/rec_att_head.py index a7cfe128..59e26c1e 100644 --- a/ppocr/modeling/heads/rec_att_head.py +++ b/ppocr/modeling/heads/rec_att_head.py @@ -146,6 +146,9 @@ 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( -- GitLab