convnext_small

holocron.models.convnext_small(pretrained: bool = False, checkpoint: Checkpoint | None = None, progress: bool = True, **kwargs: Any) → ConvNeXt

ConvNeXt-S from “A ConvNet for the 2020s”

Parameters:

• pretrained (bool) – If True, returns a model pre-trained on ImageNette

• checkpoint – If specified, the model’s parameters will be set to the checkpoint’s values

• progress (bool) – If True, displays a progress bar of the download to stderr

• kwargs – keyword args of _convnext

Returns:

classification model

Return type:

torch.nn.Module