Skip to content

Commit 3fac2e6

Browse files
added kwargs to the model() method (airctic#883)
1 parent d731a1c commit 3fac2e6

File tree

1 file changed

+2
-0
lines changed
  • icevision/models/ross/efficientdet

1 file changed

+2
-0
lines changed

icevision/models/ross/efficientdet/model.py

+2
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ def model(
1313
backbone: EfficientDetBackboneConfig,
1414
num_classes: int,
1515
img_size: int,
16+
**kwargs,
1617
) -> nn.Module:
1718
"""Creates the efficientdet model specified by `model_name`.
1819
@@ -39,6 +40,7 @@ def model(
3940
bench_labeler=True,
4041
num_classes=num_classes - 1,
4142
pretrained=backbone.pretrained,
43+
**kwargs,
4244
)
4345

4446
# TODO: Break down param groups for backbone

0 commit comments

Comments
 (0)