Module contrib.mae

Sub-modules

contrib.mae.example
contrib.mae.modeling

Functions

def load_ckpt(ckpt: str, *, chunk_size_kb: int = 1024) ‑> MaskedAutoencoder

Loads a pre-trained MAE ViT from disk. If it's not on disk, downloads the checkpoint from huggingface and then loads it into the MaskedAutoencoder class.