unifying data types in the SLAHead (#13276)

PaddlePaddle · Jul 6, 2024 · 024da2a · 024da2a
1 parent 153de46
commit 024da2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppocr/modeling/heads/table_att_head.py b/ppocr/modeling/heads/table_att_head.py
@@ -368,7 +368,7 @@ def forward(self, inputs, targets=None):
             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)