resnet152

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

ResNet-152 from “Deep Residual Learning for Image Recognition”

Parameters:

• pretrained – If True, returns a model pre-trained on ImageNet

• checkpoint – If specified, load that checkpoint on the model

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

• kwargs – keyword args of _resnet

Returns:

classification model

Return type:

torch.nn.Module