Skip to content

Commit 99ba5d9

Browse files
committed
scale reg loss for better regression
1 parent 7b94f01 commit 99ba5d9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

efficientdet/loss.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,4 +178,4 @@ def forward(self, classifications, regressions, anchors, annotations, **kwargs):
178178
display(out, imgs, obj_list, imshow=False, imwrite=True)
179179

180180
return torch.stack(classification_losses).mean(dim=0, keepdim=True), \
181-
torch.stack(regression_losses).mean(dim=0, keepdim=True)
181+
torch.stack(regression_losses).mean(dim=0, keepdim=True) * 50 # https://github.com/google/automl/blob/6fdd1de778408625c1faf368a327fe36ecd41bf7/efficientdet/hparams_config.py#L233

0 commit comments

Comments
 (0)