densenet201

torchvision.models.densenet201(pretrained: bool = False, progress: bool = True, **kwargs: Any) → torchvision.models.densenet.DenseNet

Densenet-201 model from “Densely Connected Convolutional Networks”. The required minimum input size of the model is 29x29.

Parameters

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

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

• memory_efficient (bool) – but slower. Default: False. See “paper”.