mobilevitv2

Model type

mobilevitv2

Class

ViTModel

Task

image-classification

Source

models/vit.py

Description

Vision Transformer for image feature extraction.

Usage

mobius build --model <MODEL_ID> output_dir/
from mobius import build

model = build("<MODEL_ID>")