unifying data types in the SLAHead (#13276)

GreatV · web-flow · commit 024da2a58c6e · 2024-07-06T22:01:06.000+08:00
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)

Original file line number	Diff line number	Diff line change
`@@ -368,7 +368,7 @@ def forward(self, inputs, targets=None):`
`368`	`368`	`loc_preds = loc_preds[:, : max_len + 1]`
`369`	`369`	`else:`
`370`	`370`	`structure_ids = paddle.zeros(`
`371`		`- (batch_size, self.max_text_length + 1), dtype=paddle.int64`
	`371`	`+ (batch_size, self.max_text_length + 1), dtype="int32"`
`372`	`372`	`)`
`373`	`373`	`pre_chars = paddle.zeros(shape=[batch_size], dtype="int32")`
`374`	`374`	`max_text_length = paddle.to_tensor(self.max_text_length)`