Merge pull request #2228 from WenmuZhou/fix_attn_export

fix attn export

Merge pull request #2228 from WenmuZhou/fix_attn_export
fix attn export
ddbba2b1 · zhoujun · GitHub · f687e092 · 592c4187 · ddbba2b1
Unverified Commit ddbba2b1 authored Mar 16, 2021 by zhoujun Committed by GitHub Mar 16, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

ppocr/modeling/heads/rec_att_head.py ppocr/modeling/heads/rec_att_head.py +1 -1

No files found.
--- a/ppocr/modeling/heads/rec_att_head.py
+++ b/ppocr/modeling/heads/rec_att_head.py
@@ -38,7 +38,7 @@ class AttentionHead(nn.Layer):
        return input_ont_hot

    def forward(self, inputs, targets=None, batch_max_length=25):
-        batch_size = inputs.shape[0]
+        batch_size = paddle.shape(inputs)[0]
        num_steps = batch_max_length

        hidden = paddle.zeros((batch_size, self.hidden_size))