Remove redundant `ComputeLoss` code (#4701)
Browse files- utils/loss.py +0 -1
utils/loss.py
CHANGED
|
@@ -91,7 +91,6 @@ class QFocalLoss(nn.Module):
|
|
| 91 |
class ComputeLoss:
|
| 92 |
# Compute losses
|
| 93 |
def __init__(self, model, autobalance=False):
|
| 94 |
-
super(ComputeLoss, self).__init__()
|
| 95 |
self.sort_obj_iou = False
|
| 96 |
device = next(model.parameters()).device # get model device
|
| 97 |
h = model.hyp # hyperparameters
|
|
|
|
| 91 |
class ComputeLoss:
|
| 92 |
# Compute losses
|
| 93 |
def __init__(self, model, autobalance=False):
|
|
|
|
| 94 |
self.sort_obj_iou = False
|
| 95 |
device = next(model.parameters()).device # get model device
|
| 96 |
h = model.hyp # hyperparameters
|