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)