fix bug in iou loss style (#713)

4c5cfaea · Thang Vu · Kai Chen · 2b8ad3e0 · 4c5cfaea
Commit 4c5cfaea authored May 25, 2019 by Thang Vu Committed by Kai Chen May 25, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

mmdet/models/losses/iou_loss.py mmdet/models/losses/iou_loss.py +1 -0

No files found.
--- 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,