diff --git a/ppdet/modeling/post_process.py b/ppdet/modeling/post_process.py index 01080c8fff71a0d872efec5879ac21feb5691932..6cbb079d60088c63641178666b29989819681b6f 100644 --- a/ppdet/modeling/post_process.py +++ b/ppdet/modeling/post_process.py @@ -251,7 +251,7 @@ class S2ANetBBoxPostProcess(nn.Layer): def __init__(self, num_classes=15, nms_pre=2000, min_bbox_size=0, nms=None): super(S2ANetBBoxPostProcess, self).__init__() self.num_classes = num_classes - self.nms_pre = paddle.to_tensor(nms_pre) + self.nms_pre = nms_pre self.min_bbox_size = min_bbox_size self.nms = nms self.origin_shape_list = []