convnext_small¶
-
torchvision.models.
convnext_small
(*, pretrained: bool = False, progress: bool = True, **kwargs: Any) → torchvision.models.convnext.ConvNeXt[source]¶ ConvNeXt Small model architecture from the “A ConvNet for the 2020s” paper. :param pretrained: If True, returns a model pre-trained on ImageNet :type pretrained: bool :param progress: If True, displays a progress bar of the download to stderr :type progress: bool