From 54f2411f2bd8d26b2029709a02ae6e9a884c2f9f Mon Sep 17 00:00:00 2001
From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com>
Date: Mon, 27 Dec 2021 22:53:38 +0800
Subject: [PATCH] modify infer problem of s2anet (#5010)

---
 ppdet/modeling/post_process.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppdet/modeling/post_process.py b/ppdet/modeling/post_process.py
index 765e05fbf..967d43db0 100644
--- a/ppdet/modeling/post_process.py
+++ b/ppdet/modeling/post_process.py
@@ -252,7 +252,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 = []
-- 
GitLab