We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7b94f01 commit 99ba5d9Copy full SHA for 99ba5d9
efficientdet/loss.py
@@ -178,4 +178,4 @@ def forward(self, classifications, regressions, anchors, annotations, **kwargs):
178
display(out, imgs, obj_list, imshow=False, imwrite=True)
179
180
return torch.stack(classification_losses).mean(dim=0, keepdim=True), \
181
- torch.stack(regression_losses).mean(dim=0, keepdim=True)
+ 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