Remove redundant ComputeLoss code (#4701)

This commit is contained in:
Zhiqiang Wang authored and GitHub committed 2021-09-08 12:48:33 +02:00
1 parent 2317f86ca4
commit 8e5f9ddbdb
1 file changed
-1
-1
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