diff --git a/mmdet/models/losses/iou_loss.py b/mmdet/models/losses/iou_loss.py index 8c9d602f0b72f3c48b70a99ed62cb2f8555528fa..edff11b0eb1b0f71d7a49278b209fe36c0bae3f9 100644 --- a/mmdet/models/losses/iou_loss.py +++ b/mmdet/models/losses/iou_loss.py @@ -19,6 +19,7 @@ class IoULoss(nn.Module): pred, target, weight, + style=self.style, beta=self.beta, eps=self.eps, *args,