veb-101/apple-fastvit-Keras-3

Please check out my github repo: https://github.com/veb-101/keras-vision for more models

Install PyPI package

pip install -U keras-vision

pip install git+https://github.com/veb-101/keras-vision.git

import os

os.environ["KERAS_BACKEND"] = "jax"
# os.environ["KERAS_BACKEND"] = "tensorflow"
# os.environ["KERAS_BACKEND"] = "torch"

from keras_vision.fastvit import fastvit_ma36

model = fastvit_ma36(
    pretrained=True,
    inference_mode=True,
    load_kd_weights=True,
)
model.summary()