fix MaskMatrixNMS register (#2111)

9a2651fe · Guanghua Yu · GitHub · 2f709dd1 · 9a2651fe
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

dygraph/ppdet/modeling/layers.py dygraph/ppdet/modeling/layers.py +1 -0

未找到文件。
--- a/dygraph/ppdet/modeling/layers.py
+++ b/dygraph/ppdet/modeling/layers.py
@@ -994,6 +994,7 @@ class TTFBox(object):
        return results, paddle.shape(results)[0:1]


+@register
 @serializable
 class MaskMatrixNMS(object):
    """