Skip to content

Commit

Permalink
Remove redundant ComputeLoss code (ultralytics#4701)
Browse files Browse the repository at this point in the history
  • Loading branch information
zhiqwang committed Sep 8, 2021
1 parent 64fc648 commit 8f23f05
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion utils/loss.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,6 @@ def forward(self, pred, true):
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
Expand Down

0 comments on commit 8f23f05

Please sign in to comment.