unifying data types in the SLAHead (#13276)

2024-07-06 22:01:06 +08:00 · 2024-07-06 22:01:06 +08:00 · 024da2a58c
parent 153de46b67
commit 024da2a58c
1 changed files with 1 additions and 1 deletions
--- a/ppocr/modeling/heads/table_att_head.py
+++ b/ppocr/modeling/heads/table_att_head.py
@ -368,7 +368,7 @@ class SLAHead(nn.Layer):
            loc_preds = loc_preds[:, : max_len + 1]
        else:
            structure_ids = paddle.zeros(
-                (batch_size, self.max_text_length + 1), dtype=paddle.int64
+                (batch_size, self.max_text_length + 1), dtype="int32"
            )
            pre_chars = paddle.zeros(shape=[batch_size], dtype="int32")
            max_text_length = paddle.to_tensor(self.max_text_length)