model:
  name: convmixer-768-32
  display_name: ConvMixer-768/32
  upstream: timm/convmixer_768_32.in1k
  source:
    kind: timm
    id: convmixer_768_32.in1k
    url: https://huggingface.co/timm/convmixer_768_32.in1k
    note: >-
      Original checkpoint from the paper authors ("Patches Are All You
      Need?", Trockman & Kolter, 2022, arXiv:2201.09792), original
      implementation at https://github.com/locuslab/convmixer. Confirmed
      via compile-artifact filename
      (convmixer_768_32_quantization_config.json) in the source benchmark
      export.

architecture:
  family: convmixer
  variant: 768-32
  dataset: imagenet-1k
  input_resolution: 224x224
  num_classes: 1000
  modality:
    - vision
  parameters: 21.1M
  parameters_source: published
  parameters_note: >-
    timm model card stats (convmixer_768_32.in1k), Params (M): 21.1,
    GMACs: 19.5; original paper reports ~21M params, 80.2% top-1 on
    ImageNet for ConvMixer-768/32.

format:
  type: onnx

tasks:
  - image-classification