Skip to content

Commit 09c92bb

Browse files
committed
fix: Fixed typing annotations & default values
1 parent 9dd0ce7 commit 09c92bb

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

torchvision/models/segmentation/segmentation.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def _segm_model(
2828
name: str,
2929
backbone_name: str,
3030
num_classes: int,
31-
aux: Optional[bool] = None,
31+
aux: bool,
3232
pretrained_backbone: bool = True
3333
) -> nn.Module:
3434
if 'resnet' in backbone_name:
@@ -80,7 +80,7 @@ def _load_model(
8080
pretrained: bool,
8181
progress: bool,
8282
num_classes: int,
83-
aux_loss: Optional[bool] = None,
83+
aux_loss: bool,
8484
**kwargs: Any
8585
) -> nn.Module:
8686
if pretrained:
@@ -218,7 +218,6 @@ def lraspp_mobilenet_v3_large(
218218
pretrained: bool = False,
219219
progress: bool = True,
220220
num_classes: int = 21,
221-
aux_loss: Optional[bool] = None,
222221
**kwargs: Any
223222
) -> nn.Module:
224223
"""Constructs a Lite R-ASPP Network model with a MobileNetV3-Large backbone.

0 commit comments

Comments
 (0)