From 8e5f9ddbdb3375d00483db1616ce891886072055 Mon Sep 17 00:00:00 2001 From: Zhiqiang Wang Date: Wed, 8 Sep 2021 18:48:33 +0800 Subject: [PATCH] Remove redundant `ComputeLoss` code (#4701) --- utils/loss.py | 1 - 1 file changed, 1 deletion(-) diff --git a/utils/loss.py b/utils/loss.py index 29aac31..fac432d 100644 --- a/utils/loss.py +++ b/utils/loss.py @@ -91,7 +91,6 @@ class QFocalLoss(nn.Module): class ComputeLoss: # Compute losses def __init__(self, model, autobalance=False): - super(ComputeLoss, self).__init__() self.sort_obj_iou = False device = next(model.parameters()).device # get model device h = model.hyp # hyperparameters