Remove redundant `ComputeLoss` code (#4701)
This commit is contained in:
parent
2317f86ca4
commit
8e5f9ddbdb
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue