Remove redundant `ComputeLoss` code (#4701)

This commit is contained in:
Zhiqiang Wang 2021-09-08 18:48:33 +08:00 committed by GitHub
parent 2317f86ca4
commit 8e5f9ddbdb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions

View File

@ -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