File tree Expand file tree Collapse file tree 1 file changed +2
-3
lines changed
torchvision/models/segmentation Expand file tree Collapse file tree 1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -28,7 +28,7 @@ def _segm_model(
28
28
name : str ,
29
29
backbone_name : str ,
30
30
num_classes : int ,
31
- aux : Optional [ bool ] = None ,
31
+ aux : bool ,
32
32
pretrained_backbone : bool = True
33
33
) -> nn .Module :
34
34
if 'resnet' in backbone_name :
@@ -80,7 +80,7 @@ def _load_model(
80
80
pretrained : bool ,
81
81
progress : bool ,
82
82
num_classes : int ,
83
- aux_loss : Optional [ bool ] = None ,
83
+ aux_loss : bool ,
84
84
** kwargs : Any
85
85
) -> nn .Module :
86
86
if pretrained :
@@ -218,7 +218,6 @@ def lraspp_mobilenet_v3_large(
218
218
pretrained : bool = False ,
219
219
progress : bool = True ,
220
220
num_classes : int = 21 ,
221
- aux_loss : Optional [bool ] = None ,
222
221
** kwargs : Any
223
222
) -> nn .Module :
224
223
"""Constructs a Lite R-ASPP Network model with a MobileNetV3-Large backbone.
You can’t perform that action at this time.
0 commit comments