From 7715f2357cf0ce6b6b9db4a48d2d990bc2383dbd Mon Sep 17 00:00:00 2001
From: Jiangmiao Pang <pangjiangmiao@gmail.com>
Date: Tue, 11 Jun 2019 23:02:44 +0800
Subject: [PATCH] Fix bugs in balanced l1 loss (#790)
* Update Libra R-CNN Benchmark
* fix bug in balanced_l1_loss
---
mmdet/models/losses/balanced_l1_loss.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/mmdet/models/losses/balanced_l1_loss.py b/mmdet/models/losses/balanced_l1_loss.py
index dccb9e5..2dee674 100644
--- a/mmdet/models/losses/balanced_l1_loss.py
+++ b/mmdet/models/losses/balanced_l1_loss.py
@@ -2,7 +2,7 @@ import numpy as np
import torch
import torch.nn as nn
-from .utils import reduce_loss, weighted_loss
+from .utils import weighted_loss
from ..registry import LOSSES
@@ -23,8 +23,6 @@ def balanced_l1_loss(pred,
(b * diff + 1) * torch.log(b * diff / beta + 1) - alpha * diff,
gamma * diff + gamma / b - alpha * beta)
- loss = reduce_loss(loss, reduction)
-
return loss
--
GitLab