Sensa (from the Latin sēnsus, meaning “perception”): A PyTorch Lightning based library providing Masked Autoencoder (MAE) pretraining and classification training using customizable Vision Transformer (ViT) backbones.
- Masked Autoencoder Pretraining with random token masking and reconstruction.
- Classification Training with flexible pooling strategies (avg, full, half, token).
MIT License