zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
utils/loss.py
View File

@ -91,7 +91,6 @@ class QFocalLoss(nn.Module):
class ComputeLoss: class ComputeLoss:
# Compute losses # Compute losses
def __init__(self, model, autobalance=False): def __init__(self, model, autobalance=False):
super(ComputeLoss, self).__init__()
self.sort_obj_iou = False self.sort_obj_iou = False
device = next(model.parameters()).device # get model device device = next(model.parameters()).device # get model device
h = model.hyp # hyperparameters h = model.hyp # hyperparameters