Merge branch 'dygraph' into fix_doc

ae7e1502 · zhoujun · GitHub · 4370693f · ddbba2b1 · ae7e1502
Unverified Commit ae7e1502 authored Mar 16, 2021 by zhoujun Committed by GitHub Mar 16, 2021
Hide 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))