From 51aa02f97d4bce5cb116ddf3600bb99bce69b238 Mon Sep 17 00:00:00 2001 From: wanghuancoder Date: Tue, 1 Aug 2023 19:05:10 +0800 Subject: [PATCH] paddle support stride, fix dy2st check (#10498) --- ppocr/modeling/heads/rec_robustscanner_head.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/ppocr/modeling/heads/rec_robustscanner_head.py b/ppocr/modeling/heads/rec_robustscanner_head.py index 7956059e..550836bd 100644 --- a/ppocr/modeling/heads/rec_robustscanner_head.py +++ b/ppocr/modeling/heads/rec_robustscanner_head.py @@ -99,10 +99,11 @@ class DotProductAttentionLayer(nn.Layer): logits = paddle.reshape(logits, [n, c, h, w]) if valid_ratios is not None: # cal mask of attention weight - for i, valid_ratio in enumerate(valid_ratios): - valid_width = min(w, int(w * valid_ratio + 0.5)) - if valid_width < w: - logits[i, :, :, valid_width:] = float('-inf') + with paddle.fluid.framework._stride_in_no_check_dy2st_diff(): + for i, valid_ratio in enumerate(valid_ratios): + valid_width = min(w, int(w * valid_ratio + 0.5)) + if valid_width < w: + logits[i, :, :, valid_width:] = float('-inf') # reshape to (n, c, h, w) logits = paddle.reshape(logits, [n, c, t]) -- GitLab