fix coordconv bug

58940e97 · longxiang · 892ddd84 · 58940e97
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppdet/modeling/anchor_heads/yolo_head.py ppdet/modeling/anchor_heads/yolo_head.py +1 -1

未找到文件。
--- a/ppdet/modeling/anchor_heads/yolo_head.py
+++ b/ppdet/modeling/anchor_heads/yolo_head.py
@@ -121,7 +121,7 @@ class YOLOv3Head(object):
                 act='leaky',
                 is_test=True,
                 name=None):
-        if coord_conv:
+        if coord_conv and self.coord_conv:
            input = self._add_coord(input)
        conv = fluid.layers.conv2d(
            input=input,