We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d731a1c commit 3fac2e6Copy full SHA for 3fac2e6
icevision/models/ross/efficientdet/model.py
@@ -13,6 +13,7 @@ def model(
13
backbone: EfficientDetBackboneConfig,
14
num_classes: int,
15
img_size: int,
16
+ **kwargs,
17
) -> nn.Module:
18
"""Creates the efficientdet model specified by `model_name`.
19
@@ -39,6 +40,7 @@ def model(
39
40
bench_labeler=True,
41
num_classes=num_classes - 1,
42
pretrained=backbone.pretrained,
43
44
)
45
46
# TODO: Break down param groups for backbone
0 commit comments