Remove redundant `ComputeLoss` code (#4701) · 8e5f9ddbdb - yolov5 - 拓恒技术版本管理系统

Remove redundant `ComputeLoss` code (#4701)

modifyDataloader

Zhiqiang Wang

GitHub 3 years ago

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

Split View

							
								+ 0
								
								- 1
							
utils/loss.pyView 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