mirrors
/
PaddleOCR
mirror of https://github.com/PaddlePaddle/PaddleOCR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

paddle.shape return int64 tensor (#14318)

pull/14338/head
wanghuancoder 2024-12-04 14:26:57 +08:00 committed by GitHub
parent 83323b55d5
commit 9c01b43301
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppocr/modeling/heads/rec_sar_head.py
View File

@ -242,7 +242,7 @@ class ParallelSARDecoder(BaseDecoder):
# bsz * (seq_len + 1) * h * w * attn_size
attn_weight = self.conv1x1_2(attn_weight)
# bsz * (seq_len + 1) * h * w * 1
bsz, T, h, w, c = paddle.shape(attn_weight)
bsz, T, h, w, c = paddle.shape(attn_weight).astype("int32")
assert c == 1
if valid_ratios is not None: