Skip to content

GitLab

Unverified Commit 253cd0f7 authored by zhoujun's avatar zhoujun Committed by GitHub
Browse files

Merge pull request #2029 from WenmuZhou/fix_doc 

Fix export bug
parents 85d340bd 0507402a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
ppocr/modeling/heads/rec_att_head.py
View file @ 253cd0f7
...@@ -57,6 +57,9 @@ class AttentionHead(nn.Layer): ...@@ -57,6 +57,9 @@ class AttentionHead(nn.Layer):
else: else:
targets = paddle.zeros(shape=[batch_size], dtype="int32") targets = paddle.zeros(shape=[batch_size], dtype="int32")
probs = None probs = None
char_onehots = None
outputs = None
alpha = None
for i in range(num_steps): for i in range(num_steps):
char_onehots = self._char_to_onehot( char_onehots = self._char_to_onehot(
...@@ -146,9 +149,6 @@ class AttentionLSTM(nn.Layer): ...@@ -146,9 +149,6 @@ class AttentionLSTM(nn.Layer):
else: else:
targets = paddle.zeros(shape=[batch_size], dtype="int32") targets = paddle.zeros(shape=[batch_size], dtype="int32")
probs = None probs = None
char_onehots = None
outputs = None
alpha = None
for i in range(num_steps): for i in range(num_steps):
char_onehots = self._char_to_onehot( char_onehots = self._char_to_onehot(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment