unify int32 in op type (#3467)

6063d163 · wangguanzhong · GitHub · a8015e4b · 6063d163
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

ppdet/modeling/ops.py ppdet/modeling/ops.py +2 -2

未找到文件。
--- 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')