add repr

PaddlePaddle · Aug 24, 2021 · 3bf998e · 3bf998e · paddle-bot-old · Aug 24, 2021
1 parent b6b8311
commit 3bf998e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/paddle/vision/ops.py b/python/paddle/vision/ops.py
@@ -1024,3 +1024,7 @@ def forward(self, input, boxes, boxes_num=None):
             output_size=self._output_size,
             spatial_scale=self._spatial_scale,
             boxes_num=boxes_num)
+
+    def extra_repr(self):
+        main_str = 'output_size={_output_size}, spatial_scale={_spatial_scale}'
+        return main_str.format(**self.__dict__)