diff --git a/ppdet/modeling/architectures/blazeface.py b/ppdet/modeling/architectures/blazeface.py
index 9b292b6a9c79f840aaa17ea896a9ae6eda58b522..b8a46535e901a12fa0c23c5e51d2978d6b32ccf6 100644
--- a/ppdet/modeling/architectures/blazeface.py
+++ b/ppdet/modeling/architectures/blazeface.py
@@ -176,7 +176,7 @@ class BlazeFace(object):
         # yapf: disable
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
             'im_shape': {'shape': [None, 3], 'dtype': 'int32',   'lod_level': 0},
diff --git a/ppdet/modeling/architectures/cascade_mask_rcnn.py b/ppdet/modeling/architectures/cascade_mask_rcnn.py
index e04b9af3e5d8665446f842835b89873e08babe12..335e53cd1fdf11768a912a965b7faedca82e149d 100644
--- a/ppdet/modeling/architectures/cascade_mask_rcnn.py
+++ b/ppdet/modeling/architectures/cascade_mask_rcnn.py
@@ -371,7 +371,7 @@ class CascadeMaskRCNN(object):
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
             'im_info':  {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'im_shape': {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
diff --git a/ppdet/modeling/architectures/cascade_rcnn.py b/ppdet/modeling/architectures/cascade_rcnn.py
index 5942a445d5a2a94c522360d473f1219ff59eb231..7f57d932d588121c9fef57148984e79e85b63fad 100644
--- a/ppdet/modeling/architectures/cascade_rcnn.py
+++ b/ppdet/modeling/architectures/cascade_rcnn.py
@@ -283,7 +283,7 @@ class CascadeRCNN(object):
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
             'im_info':  {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
             'im_shape': {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
             'is_crowd': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
diff --git a/ppdet/modeling/architectures/cascade_rcnn_cls_aware.py b/ppdet/modeling/architectures/cascade_rcnn_cls_aware.py
index e52ee831e8ab50db8752bab671b5f6275e54be18..2f9696db946acdbd6dbd0e05a9a569d63da8fc69 100644
--- a/ppdet/modeling/architectures/cascade_rcnn_cls_aware.py
+++ b/ppdet/modeling/architectures/cascade_rcnn_cls_aware.py
@@ -176,7 +176,7 @@ class CascadeRCNNClsAware(object):
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
             'im_info':  {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'im_shape': {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
diff --git a/ppdet/modeling/architectures/faceboxes.py b/ppdet/modeling/architectures/faceboxes.py
index 0e7c0dec430800ddaa4e6ea28e0cdc502566c8be..ee7292017a2afa0933c79cb5a0a92e731f763f36 100644
--- a/ppdet/modeling/architectures/faceboxes.py
+++ b/ppdet/modeling/architectures/faceboxes.py
@@ -150,7 +150,7 @@ class FaceBoxes(object):
         # yapf: disable
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
             'im_shape': {'shape': [None, 3], 'dtype': 'int32',   'lod_level': 0},
diff --git a/ppdet/modeling/architectures/faster_rcnn.py b/ppdet/modeling/architectures/faster_rcnn.py
index c1730a1d784743176e3f87f4446de11a4304a60c..57a8ffb066280c2da35adaf55ea4f376111ee251 100644
--- a/ppdet/modeling/architectures/faster_rcnn.py
+++ b/ppdet/modeling/architectures/faster_rcnn.py
@@ -194,7 +194,7 @@ class FasterRCNN(object):
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
             'im_info':  {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'im_shape': {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
diff --git a/ppdet/modeling/architectures/mask_rcnn.py b/ppdet/modeling/architectures/mask_rcnn.py
index 9e068531ebb59d11f5f1c11e18cba97f64db39c9..019a4bda70774b4f6ee9d0d6b14ad75d91af2209 100644
--- a/ppdet/modeling/architectures/mask_rcnn.py
+++ b/ppdet/modeling/architectures/mask_rcnn.py
@@ -274,7 +274,7 @@ class MaskRCNN(object):
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
             'im_info':  {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'im_shape': {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
diff --git a/ppdet/modeling/architectures/retinanet.py b/ppdet/modeling/architectures/retinanet.py
index 00478da925f71106ec8a99c00287d20ded3d4c43..e4613aedfeedd0f94e5c343bb151443720205f11 100644
--- a/ppdet/modeling/architectures/retinanet.py
+++ b/ppdet/modeling/architectures/retinanet.py
@@ -88,7 +88,7 @@ class RetinaNet(object):
         inputs_def = {
             'image':    {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
             'im_info':  {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
-            'im_id':    {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'im_shape': {'shape': [None, 3], 'dtype': 'float32', 'lod_level': 0},
             'gt_bbox':  {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class': {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
diff --git a/ppdet/modeling/architectures/ssd.py b/ppdet/modeling/architectures/ssd.py
index de7dd461de702da0b58750ad0483780583a92915..70aaf7319e259195e58abe325b00bb2eae6fad2b 100644
--- a/ppdet/modeling/architectures/ssd.py
+++ b/ppdet/modeling/architectures/ssd.py
@@ -95,7 +95,7 @@ class SSD(object):
         # yapf: disable
         inputs_def = {
             'image':        {'shape': im_shape,  'dtype': 'float32', 'lod_level': 0},
-            'im_id':        {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 0},
+            'im_id':        {'shape': [None, 1], 'dtype': 'int64',   'lod_level': 0},
             'gt_bbox':      {'shape': [None, 4], 'dtype': 'float32', 'lod_level': 1},
             'gt_class':     {'shape': [None, 1], 'dtype': 'int32',   'lod_level': 1},
             'im_shape':     {'shape': [None, 3], 'dtype': 'int32',   'lod_level': 0},
diff --git a/ppdet/modeling/architectures/yolov3.py b/ppdet/modeling/architectures/yolov3.py
index e82f3b1c04999d4d8ac324e06cd95cca46e44e2f..e0efccae17547737462c45ce0f52583630ca37dc 100644
--- a/ppdet/modeling/architectures/yolov3.py
+++ b/ppdet/modeling/architectures/yolov3.py
@@ -96,7 +96,7 @@ class YOLOv3(object):
         inputs_def = {
             'image':    {'shape': im_shape,                 'dtype': 'float32', 'lod_level': 0},
             'im_size':  {'shape': [None, 2],                'dtype': 'int32',   'lod_level': 0},
-            'im_id':    {'shape': [None, 1],                'dtype': 'int32',   'lod_level': 0},
+            'im_id':    {'shape': [None, 1],                'dtype': 'int64',   'lod_level': 0},
             'gt_bbox':  {'shape': [None, num_max_boxes, 4], 'dtype': 'float32', 'lod_level': 0},
             'gt_class': {'shape': [None, num_max_boxes],    'dtype': 'int32',   'lod_level': 0},
             'gt_score': {'shape': [None, num_max_boxes],    'dtype': 'float32', 'lod_level': 0},