vgg_16_bn_cls(pretrained: bool | str = False, weights: VGG16BNWeights | None = None, overrides: object = {})VGG-16 with BatchNorm — image classifier.
Same topology as vgg_16_cls but with
lucid.nn.BatchNorm2d after each Conv + ReLU pair.
Approximately 138.4 M parameters; top-1 ImageNet validation
accuracy of 73.36% (torchvision eval). Often preferred over plain
vgg_16_cls for downstream fine-tuning.
Model Size
Parameters
pretrainedbool or str= FalsePretrained-weight selector.
False → random init; True
→ the DEFAULT tag (VGG16BNWeights.IMAGENET1K_V1); a
tag string → that specific checkpoint. Mutually exclusive with
weights (which wins if both are given).Explicit weights enum member. Takes precedence over
pretrained.**overridesobject= {}Keyword overrides forwarded into
VGGConfig.Returns
VGGForImageClassificationClassifier with the VGG-16-BN configuration applied (or with
overrides merged on top of it), optionally initialised from
pretrained weights.
Notes
Pretrained weights are converted from torchvision's
VGG16_BN_Weights and hosted on the Hugging Face Hub under
lucid-dl/vgg-16-bn.
Examples
>>> import lucid
>>> from lucid.models.vision.vgg import vgg_16_bn_cls
>>> model = vgg_16_bn_cls(pretrained=True)
>>> x = lucid.randn(1, 3, 224, 224)
>>> out = model(x)
>>> out.logits.shape
(1, 1000)