coatnet_6_cls(pretrained: bool = False, overrides: object = {})CoAtNet-6 image classifier (Dai et al., 2021).
Mixed-S3 variant. Approximately 1.5B parameters.
Model Size
Examples
>>> import lucid
>>> from lucid.models.vision.coatnet import coatnet_6_cls
>>> model = coatnet_6_cls()
>>> out = model(lucid.randn(1, 3, 224, 224))
>>> out.logits.shape
(1, 1000)