VGG-16 with BatchNorm — feature-extracting backbone.
Builds a VGG with the VGG-16 topology
arch=(2, 2, 3, 3, 3) and lucid.nn.BatchNorm2d after
each Conv + ReLU pair. Approximately 138.4 M parameters. Often
preferred over plain vgg_16 for downstream fine-tuning
because BatchNorm stabilises gradient statistics in the early conv
stack.
Model Size
Parameters
pretrainedbool= FalseReserved for future pretrained-weight loading. Currently
ignored.
**overridesobject= {}Keyword overrides forwarded into
VGGConfig.Returns
VGGBackbone with the VGG-16-BN configuration applied (or with
overrides merged on top of it).
Examples
>>> import lucid
>>> from lucid.models.vision.vgg import vgg_16_bn
>>> model = vgg_16_bn()
>>> x = lucid.randn(1, 3, 224, 224)
>>> out = model(x)
>>> out.last_hidden_state.shape
(1, 512, 7, 7)