diff --git a/ppdet/modeling/anchor_heads/yolo_head.py b/ppdet/modeling/anchor_heads/yolo_head.py index 7af4cd324b07169662614982975a2d57481c7d16..dfcec0d87f7e124550ecac6cf97299173e9a03c6 100644 --- a/ppdet/modeling/anchor_heads/yolo_head.py +++ b/ppdet/modeling/anchor_heads/yolo_head.py @@ -384,7 +384,7 @@ class YOLOv4Head(YOLOv3Head): yolo_loss=yolo_loss, iou_aware=iou_aware, iou_aware_factor=iou_aware_factor, - clip_box=clip_bbox) + clip_bbox=clip_bbox) self.spp_stage = spp_stage def _upsample(self, input, scale=2, name=None):