← all repositories

facebookresearch/mae

A PyTorch implementation of Masked Autoencoders (MAE), a self-supervised vision transformer architecture for scalable image representation learning.

8.3k stars Python Computer VisionML Frameworks
mae
Velocity · 7d
+5.1
★ / day
Trend
steady
star history

This repository provides a PyTorch re-implementation of the MAE paper on masked autoencoders for vision, originally released in TensorFlow+TPU. It includes pre-training code for self-supervised learning on images, fine-tuning scripts with pre-trained ViT checkpoints across Base/Large/Huge sizes, and an interactive visualization demo for exploring learned representations. The implementation builds on the timm library and the DeiT repository.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.