Apply suggestions from code review

Author: datumbox

torchvision/models/segmentation/segmentation.py

@@ -125,7 +125,7 @@ def fcn_resnet50(
     num_classes: int = 21,
     aux_loss: Optional[bool] = None,
     **kwargs: Any
-) -> nn.Module:
+) -> FCN:
     """Constructs a Fully-Convolutional Network model with a ResNet-50 backbone.
 
     Args:
@@ -144,7 +144,7 @@ def fcn_resnet101(
     num_classes: int = 21,
     aux_loss: Optional[bool] = None,
     **kwargs: Any
-) -> nn.Module:
+) -> FCN:
     """Constructs a Fully-Convolutional Network model with a ResNet-101 backbone.
 
     Args:
@@ -163,7 +163,7 @@ def deeplabv3_resnet50(
     num_classes: int = 21,
     aux_loss: Optional[bool] = None,
     **kwargs: Any
-) -> nn.Module:
+) -> DeepLabV3:
     """Constructs a DeepLabV3 model with a ResNet-50 backbone.
 
     Args:
@@ -182,7 +182,7 @@ def deeplabv3_resnet101(
     num_classes: int = 21,
     aux_loss: Optional[bool] = None,
     **kwargs: Any
-) -> nn.Module:
+) -> DeepLabV3:
     """Constructs a DeepLabV3 model with a ResNet-101 backbone.
 
     Args:
@@ -201,7 +201,7 @@ def deeplabv3_mobilenet_v3_large(
     num_classes: int = 21,
     aux_loss: Optional[bool] = None,
     **kwargs: Any
-) -> nn.Module:
+) -> DeepLabV3:
     """Constructs a DeepLabV3 model with a MobileNetV3-Large backbone.
 
     Args:
@@ -219,7 +219,7 @@ def lraspp_mobilenet_v3_large(
     progress: bool = True,
     num_classes: int = 21,
     **kwargs: Any
-) -> nn.Module:
+) -> LRASPP:
     """Constructs a Lite R-ASPP Network model with a MobileNetV3-Large backbone.
 
     Args: