fix MaskMatrixNMS register (PaddlePaddle#2111)

heavengate · Jan 22, 2021 · 9a2651f · 9a2651f
1 parent 2f709dd
commit 9a2651f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dygraph/ppdet/modeling/layers.py b/dygraph/ppdet/modeling/layers.py
@@ -994,6 +994,7 @@ def __call__(self, hm, wh, im_shape, scale_factor):
         return results, paddle.shape(results)[0:1]
 
 
+@register
 @serializable
 class MaskMatrixNMS(object):
     """