updates

2019-08-18 02:04:49 +02:00
parent 43230c48bf
commit ce0b414677
2 changed files with 2 additions and 3 deletions
@@ -312,7 +312,7 @@ class FocalLoss(nn.Module):
            return loss


-def compute_loss(p, targets, model, giou_loss=True):  # predictions, targets, model
+def compute_loss(p, targets, model):  # predictions, targets, model
    ft = torch.cuda.FloatTensor if p[0].is_cuda else torch.Tensor
    lcls, lbox, lobj = ft([0]), ft([0]), ft([0])
    tcls, tbox, indices, anchor_vec = build_targets(model, targets)