vgg16_bn

torchvision.models.vgg16_bn(pretrained: bool = False, progress: bool = True, **kwargs: Any) → torchvision.models.vgg.VGG

VGG 16-layer model (configuration “D”) with batch normalization “Very Deep Convolutional Networks For Large-Scale Image Recognition”. The required minimum input size of the model is 32x32.

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