From 4c5cfaea8363b6f40579ea9163f3ca053d563b25 Mon Sep 17 00:00:00 2001
From: Thang Vu <thangvubk@gmail.com>
Date: Sat, 25 May 2019 17:09:19 +0900
Subject: [PATCH] fix bug in iou loss style (#713)
---
mmdet/models/losses/iou_loss.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/mmdet/models/losses/iou_loss.py b/mmdet/models/losses/iou_loss.py
index 8c9d602..edff11b 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,
--
GitLab