Skip to content

Commit 75e16c2

Browse files
authored
Merge pull request #208 from MichaelMonashev/patch-1
allow to override 'max_det_per_image'
2 parents b3dcb20 + 7be9ea4 commit 75e16c2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

effdet/factory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def create_model_from_config(
2323
pretrained_backbone = False # no point in loading backbone weights
2424

2525
# Config overrides, override some config values via kwargs.
26-
overrides = ('redundant_bias', 'label_smoothing', 'legacy_focal', 'jit_loss', 'soft_nms')
26+
overrides = ('redundant_bias', 'label_smoothing', 'legacy_focal', 'jit_loss', 'soft_nms', 'max_det_per_image')
2727
for ov in overrides:
2828
value = kwargs.pop(ov, None)
2929
if value is not None:

0 commit comments

Comments
 (0)