coatnet_2_cls(pretrained: bool = False, overrides: object = {})CoAtNet-2 image classifier (Dai et al., 2021).
Wider variant — dims=(128, 256, 512, 1024) — with the standard
CoAtNet head. Approximately 75M parameters, 84.1% ImageNet-1k
top-1 at 224x224 (Table 5).
Model Size
Parameters
pretrainedbool= FalseNo checkpoint is published for
coatnet_2_cls, so True raises
NotImplementedError rather than return randomly initialised weights.**overridesobject= {}Keyword overrides applied on top of the canonical CoAtNet-2
config; each must name a field of
CoAtNetConfig (e.g.
num_classes=10 to retarget the head).Examples
>>> import lucid
>>> from lucid.models.vision.coatnet import coatnet_2_cls
>>> model = coatnet_2_cls()
>>> out = model(lucid.randn(1, 3, 224, 224))
>>> out.logits.shape
(1, 1000)