Index
Trainers API
This module contains the implementations of various training routines that fall outside the standard supervised learning paradigm. These include contrastive learning, distillation, and more.
Available Trainers
- ContrastiveTrainer: A trainer for contrastive learning
- Distiller: A trainer for distillation
- GSAutoencoder: A trainer for Gumbel Softmax autoencoder
- MaskedAutoencoder: A trainer for masked autoencoder
- SimCLRTrainer: A trainer for SimCLR
- VQAutoencoder: A trainer for Vector Quantized autoencoder