fix softnms type error (#1021)

ba0303eb · Guanghua Yu · GitHub · db5f0d70 · ba0303eb
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

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

未找到文件。
--- a/ppdet/modeling/ops.py
+++ b/ppdet/modeling/ops.py
@@ -610,7 +610,7 @@ class MultiClassSoftNMS(object):
            res.set_lod([out_offsets])
            if len(pred_res) == 0:
                pred_res = np.array([[1]], dtype=np.float32)
-            res.set(np.vstack(pred_res), fluid.CPUPlace())
+            res.set(np.vstack(pred_res).astype(np.float32), fluid.CPUPlace())
            return res

        pred_result = create_tmp_var(