From e09d1c41d606161a4765d22ac46c0809eff69170 Mon Sep 17 00:00:00 2001 From: frgfm Date: Thu, 29 Jul 2021 19:09:49 +0200 Subject: [PATCH] docs: Added missing arg description in lraspp --- torchvision/models/segmentation/segmentation.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torchvision/models/segmentation/segmentation.py b/torchvision/models/segmentation/segmentation.py index 7b3a0258ddb..5b2c8a63a61 100644 --- a/torchvision/models/segmentation/segmentation.py +++ b/torchvision/models/segmentation/segmentation.py @@ -181,6 +181,7 @@ def lraspp_mobilenet_v3_large(pretrained=False, progress=True, num_classes=21, * contains the same classes as Pascal VOC progress (bool): If True, displays a progress bar of the download to stderr num_classes (int): number of output classes of the model (including the background) + aux_loss (bool): If True, it uses an auxiliary loss """ if kwargs.pop("aux_loss", False): raise NotImplementedError('This model does not use auxiliary loss')