From 6063d163aa84eea09323bdab53f52704011a69a0 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Tue, 22 Jun 2021 19:16:01 +0800 Subject: [PATCH] unify int32 in op type (#3467) --- ppdet/modeling/ops.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppdet/modeling/ops.py b/ppdet/modeling/ops.py index 55451bab8..0b9cccf9b 100644 --- a/ppdet/modeling/ops.py +++ b/ppdet/modeling/ops.py @@ -993,7 +993,7 @@ def multiclass_nms(bboxes, else: output = helper.create_variable_for_type_inference(dtype=bboxes.dtype) - index = helper.create_variable_for_type_inference(dtype='int') + index = helper.create_variable_for_type_inference(dtype='int32') inputs = {'BBoxes': bboxes, 'Scores': scores} outputs = {'Out': output, 'Index': index} @@ -1136,7 +1136,7 @@ def matrix_nms(bboxes, else: helper = LayerHelper('matrix_nms', **locals()) output = helper.create_variable_for_type_inference(dtype=bboxes.dtype) - index = helper.create_variable_for_type_inference(dtype='int') + index = helper.create_variable_for_type_inference(dtype='int32') outputs = {'Out': output, 'Index': index} if return_rois_num: rois_num = helper.create_variable_for_type_inference(dtype='int32') -- GitLab