fastvit_t8

Converted TIMM image classification model for LiteRT.

• Source architecture: fastvit_t8
• Source checkpoint: timm/fastvit_t8.apple_in1k
• File: model.tflite
• Input: float32 tensor in NCHW layout, shape [1, 3, 256, 256]
• Output: ImageNet-1K logits, shape [1, 1000]

Model Details

• Model Type: Image classification / feature backbone
• Model Stats:
  • Params (M): 4.0
  • GMACs: 0.7
  • Activations (M): 8.6
  • Image size: 256 x 256
• Papers:
  • FastViT: A Fast Hybrid Vision Transformer using Structural Reparameterization: https://arxiv.org/abs/2303.14189
• Original: https://github.com/apple/ml-fastvit
• Dataset: ImageNet-1k

Citation

@inproceedings{vasufastvit2023,
  author = {Pavan Kumar Anasosalu Vasu and James Gabriel and Jeff Zhu and Oncel Tuzel and Anurag Ranjan},
  title = {FastViT:  A Fast Hybrid Vision Transformer using Structural Reparameterization},
  booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
  year = {2023}
}