sk_resnet_34

→SKNet

sk_resnet_34(pretrained: bool = False, overrides: object = {})

source edit

SK-ResNet-34 feature-extracting backbone (no classification head).

Builds an SKNet with ResNet-34 topology (_SelectiveKernelBasic blocks stacked [3, 4, 6, 3]). Both $3 \times 3$ convolutions inside every basic block are replaced by Selective Kernel units. Approximately 46.9M parameters.

Model Size

Parameters

pretrainedbool= False

Reserved for future pretrained-weight loading. Currently ignored.

**overridesobject= {}

Keyword overrides forwarded into SKNetConfig.

Returns

SKNet

Backbone with the SK-ResNet-34 configuration applied (or with overrides merged on top of it).

Notes

See Li et al., "Selective Kernel Networks", CVPR 2019 (arXiv:1903.06586).

Examples

>>> import lucid
>>> from lucid.models.vision.sknet import sk_resnet_34
>>> model = sk_resnet_34()
>>> x = lucid.randn(1, 3, 224, 224)
>>> out = model(x)
>>> out.last_hidden_state.shape
(1, 512, 7, 7)

Used by 1

lucid.models.vision.sknet

>>> import lucid >>> from lucid.models.vision.sknet import sk_resnet_34 >>> model = sk_resnet_34() >>> x = lucid.randn(1, 3, 224, 224) >>> out = model(x) >>> out.last_hidden_state.shape (1, 512, 7, 7)