nn.Module warpper for focal loss
| 111 | |
| 112 | |
| 113 | class FocalLoss(nn.Module): |
| 114 | """nn.Module warpper for focal loss""" |
| 115 | |
| 116 | def __init__(self): |
| 117 | super(FocalLoss, self).__init__() |
| 118 | self.neg_loss = _neg_loss |
| 119 | |
| 120 | def forward(self, out, target): |
| 121 | return self.neg_loss(out, target) |
| 122 | |
| 123 | |
| 124 | class RegLoss(nn.Module): |