From 46a6950e7a6e1f9eec558e0563306cd19bc00c20 Mon Sep 17 00:00:00 2001 From: zxcd <228587199@qq.com> Date: Fri, 2 Jun 2023 14:21:17 +0800 Subject: [PATCH] add dtype param for arange API. (#10075) --- ppocr/modeling/transforms/tbsrn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppocr/modeling/transforms/tbsrn.py b/ppocr/modeling/transforms/tbsrn.py index e06ba907db..e3e77bd36a 100644 --- a/ppocr/modeling/transforms/tbsrn.py +++ b/ppocr/modeling/transforms/tbsrn.py @@ -46,7 +46,7 @@ def positionalencoding2d(d_model, height, width): # Each dimension use half of d_model d_model = int(d_model / 2) div_term = paddle.exp( - paddle.arange(0., d_model, 2) * -(math.log(10000.0) / d_model)) + paddle.arange(0., d_model, 2, dtype='int64') * -(math.log(10000.0) / d_model)) pos_w = paddle.arange(0., width, dtype='float32').unsqueeze(1) pos_h = paddle.arange(0., height, dtype='float32').unsqueeze(1)