Lightning-Bolts documentation

Start here

• Installation
• Introduction Guide

Callbacks

• Monitoring Callbacks
  • Print Table Metrics
  • Data Monitoring in LightningModule
  • Model Verification
• Torch ORT Callback
• SparseML Callback
  • 1. Choose your Sparse Recipe
  • 2. Train with SparseMLCallback
  • 3. Export to ONNX!
• Self-supervised Callbacks
  • BYOLMAWeightUpdate
  • SSLOnlineEvaluator
• Variational Callbacks
  • Latent Dim Interpolator
• Vision Callbacks
  • Confused Logit
  • Tensorboard Image Generator

DataModules

• Sklearn Datamodule
  • Sklearn Dataset Class
  • Sklearn DataModule Class
• Vision DataModules
  • Supervised learning
  • Semi-supervised learning

Datasets

• Debug Datasets
  • DummyDataset
  • DummyDetectionDataset
  • RandomDataset
  • RandomDictDataset
  • RandomDictStringDataset

DataLoaders

• AsynchronousLoader

Losses

• Object Detection
  • GIoU Loss
  • IoU Loss
• Reinforcement Learning
  • DQN Loss
  • Double DQN Loss
  • Per DQN Loss

Models

• How to use models
  • Predicting on your data
  • Finetuning on your data
  • Train from scratch
  • For research
• Autoencoders
  • Basic AE
  • Basic VAE
• Convolutional Architectures
  • GPT-2
  • Image GPT
  • Pixel CNN
  • UNet
  • Semantic Segmentation
• GANs
  • Basic GAN
  • DCGAN
  • SRGAN
• Object Detection
  • Faster R-CNN
  • RetinaNet
  • YOLO
• Reinforcement Learning
  • Module authors
  • DQN Models
  • Policy Gradient Models
  • Actor-Critic Models
• Self-supervised Learning
  • Use cases
  • Contrastive Learning Models
• Classic ML Models
  • Linear Regression
  • Logistic Regression

Learning Rate Schedulers

• Linear Warmup Cosine Annealing

Data Processing

• Self-supervised learning
  • CPC transforms
  • AMDIM transforms
  • MOCO V2 transforms
  • SimCLR transforms
  • Identity class
  • SSL-ready resnets
  • SSL backbone finetuner
• Semi-supervised learning
  • Balanced classes
  • half labeled batches

Tasks

• Self-supervised Learning
  • FeatureMapContrastiveTask
  • Context prediction tasks

Community

• Contributing
• PL Bolts Governance | Persons of interest
• Bolts stability
• Changelog

Indices and tables

• Index

• Module Index

• Search Page