Unverified 提交 8e5f9ddb authored 作者: Zhiqiang Wang's avatar Zhiqiang Wang 提交者: GitHub

Remove redundant `ComputeLoss` code (#4701)

上级 2317f86c
......@@ -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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论