Skip to content

Commit

Permalink
Fix bugs in balanced l1 loss (#790)
Browse files Browse the repository at this point in the history
* Update Libra R-CNN Benchmark

* fix bug in balanced_l1_loss
  • Loading branch information
OceanPang authored and hellock committed Jun 11, 2019
1 parent 7a6306f commit 7715f23
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions mmdet/models/losses/balanced_l1_loss.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
import torch
import torch.nn as nn

from .utils import reduce_loss, weighted_loss
from .utils import weighted_loss
from ..registry import LOSSES


Expand All @@ -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


Expand Down

0 comments on commit 7715f23

Please sign in to comment.