sknet101

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

SKNet-101 from “Selective Kernel Networks”

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

  • 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 _sknet

Returns:

classification model

Return type:

torch.nn.Module