Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Remove redundant ComputeLoss code (#4701)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
utils/loss.py
-1
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