convnext_femto

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

ConvNeXt-Femto variant of Ross Wightman inspired by “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