efficientnet_b6

→EfficientNet

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

source edit

EfficientNet-B6 feature-extracting backbone (compound coefficient $\phi = 5$ ).

Builds an EfficientNet with width multiplier 1.8 and depth multiplier 2.6. Approximately 43M parameters and 84.0% ImageNet-1k top-1 accuracy at the 528×528 native resolution (Tan & Le, 2019, Table 2).

Model Size

Parameters

pretrainedbool= False

Reserved for future pretrained-weight loading. Currently ignored.

**overridesobject= {}

Keyword overrides forwarded into EfficientNetConfig.

Returns

EfficientNet

Backbone with the B6 configuration applied (or with overrides merged on top of it).

Notes

See Tan & Le, "EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks", ICML 2019 (arXiv:1905.11946), Table 2. Head expansion: round(1280 · 1.8) = 2304 channels.

Examples

>>> import lucid
>>> from lucid.models.vision.efficientnet import efficientnet_b6
>>> model = efficientnet_b6()
>>> x = lucid.randn(1, 3, 528, 528)
>>> out = model(x)
>>> out.last_hidden_state.shape
(1, 2304, 1, 1)

Used by 1

lucid.models.vision.efficientnet

>>> import lucid >>> from lucid.models.vision.efficientnet import efficientnet_b6 >>> model = efficientnet_b6() >>> x = lucid.randn(1, 3, 528, 528) >>> out = model(x) >>> out.last_hidden_state.shape (1, 2304, 1, 1)