Hide navigation sidebar

Hide table of contents sidebar

Toggle site navigation sidebar

Toggle table of contents sidebar

Getting Started

💾 Installation
🎨 Using Composer
🚌 Welcome Tour

Trainer

⚙️ Using the Trainer
💿 Dataloaders
📊 Evaluation
📉 Schedulers
⌛ Time
📆 Events
✅ Checkpointing
🪵 Logging
☎️ Callbacks
👨‍👩‍👧‍👦 Distributed Training
🔢 Numerics
🏎️ Performance

Key Classes

🛻 ComposerModel
🤖 Algorithms

Methods Library

🃏 Methods Overview
🥸 ALiBi
🎨 AugMix
🏊 BlurPool
📺 Channels Last
🎃 CutMix
🏛️ ColOut
✂️ Cutout
🏋️‍♀️ Decoupled Weight Decay
➗ Factorize
👻 Ghost BatchNorm
🧈 Label Smoothing
❄️ Layer Freezing
🥣 MixUp
🏞️ Progressive Image Resizing
🎲 RandAugment
⚖️ Scale Schedule
⚖️ Scaling Laws
⏮️ Selective Backprop
🔆 Sequence Length Warmup
🏔️ Sharpness Aware Minimization (SAM)
🫀 Squeeze-and-Excitation
🧊 Stochastic Depth (Block-Wise)
🎰 Stochastic Depth (Sample-Wise)
🧩 Stochastic Weight Averaging

Model Library

👀 CIFAR ResNet
🕸️ EfficientNet
📚 GPT-2
🏙️ ResNet
↩️ UNet

API Reference

🎱 API Reference
Toggle child pages in navigation
- composer.algorithms
  Toggle child pages in navigation
  - composer.algorithms.algorithm_hparams
  - composer.algorithms.algorithm_registry
  - composer.algorithms.alibi
    Toggle child pages in navigation
    - composer.algorithms.alibi.alibi
  - composer.algorithms.augmix
    Toggle child pages in navigation
    - composer.algorithms.augmix.augmix
  - composer.algorithms.blurpool
    Toggle child pages in navigation
    - composer.algorithms.blurpool.blurpool
    - composer.algorithms.blurpool.blurpool_layers
  - composer.algorithms.channels_last
    Toggle child pages in navigation
    - composer.algorithms.channels_last.channels_last
  - composer.algorithms.colout
    Toggle child pages in navigation
    - composer.algorithms.colout.colout
  - composer.algorithms.cutmix
    Toggle child pages in navigation
    - composer.algorithms.cutmix.cutmix
  - composer.algorithms.cutout
    Toggle child pages in navigation
    - composer.algorithms.cutout.cutout
  - composer.algorithms.factorize
    Toggle child pages in navigation
  - composer.algorithms.ghost_batchnorm
    Toggle child pages in navigation
    - composer.algorithms.ghost_batchnorm.ghost_batchnorm
  - composer.algorithms.hparams
  - composer.algorithms.label_smoothing
    Toggle child pages in navigation
    - composer.algorithms.label_smoothing.label_smoothing
  - composer.algorithms.layer_freezing
    Toggle child pages in navigation
    - composer.algorithms.layer_freezing.layer_freezing
  - composer.algorithms.mixup
    Toggle child pages in navigation
    - composer.algorithms.mixup.mixup
  - composer.algorithms.no_op_model
    Toggle child pages in navigation
    - composer.algorithms.no_op_model.no_op_model
  - composer.algorithms.progressive_resizing
    Toggle child pages in navigation
    - composer.algorithms.progressive_resizing.progressive_resizing
  - composer.algorithms.randaugment
    Toggle child pages in navigation
    - composer.algorithms.randaugment.randaugment
  - composer.algorithms.sam
    Toggle child pages in navigation
    - composer.algorithms.sam.sam
  - composer.algorithms.scale_schedule
    Toggle child pages in navigation
    - composer.algorithms.scale_schedule.scale_schedule
  - composer.algorithms.selective_backprop
    Toggle child pages in navigation
    - composer.algorithms.selective_backprop.selective_backprop
  - composer.algorithms.seq_length_warmup
    Toggle child pages in navigation
    - composer.algorithms.seq_length_warmup.seq_length_warmup
  - composer.algorithms.squeeze_excite
    Toggle child pages in navigation
    - composer.algorithms.squeeze_excite.squeeze_excite
  - composer.algorithms.stochastic_depth
    Toggle child pages in navigation
  - composer.algorithms.swa
    Toggle child pages in navigation
    - composer.algorithms.swa.swa
  - composer.algorithms.utils
    Toggle child pages in navigation
    - composer.algorithms.utils.augmentation_common
    - composer.algorithms.utils.augmentation_primitives
- composer.callbacks
  Toggle child pages in navigation
- composer.core
  Toggle child pages in navigation
  - composer.core.algorithm
  - composer.core.callback
  - composer.core.data_spec
  - composer.core.engine
  - composer.core.evaluator
  - composer.core.event
  - composer.core.logging
    Toggle child pages in navigation
    - composer.core.logging.base_backend
    - composer.core.logging.logger
  - composer.core.precision
  - composer.core.serializable
  - composer.core.state
  - composer.core.time
  - composer.core.types
- composer.datasets
  Toggle child pages in navigation
- composer.functional
- composer.loggers
  Toggle child pages in navigation
- composer.models
  Toggle child pages in navigation
  - composer.models.base
  - composer.models.bert
    Toggle child pages in navigation
    - composer.models.bert.bert_hparams
    - composer.models.bert.model
  - composer.models.classify_mnist
    Toggle child pages in navigation
    - composer.models.classify_mnist.mnist_hparams
    - composer.models.classify_mnist.model
  - composer.models.deeplabv3
    Toggle child pages in navigation
    - composer.models.deeplabv3.deeplabv3
    - composer.models.deeplabv3.deeplabv3_hparams
  - composer.models.efficientnetb0
    Toggle child pages in navigation
    - composer.models.efficientnetb0.efficientnetb0_hparams
    - composer.models.efficientnetb0.model
  - composer.models.efficientnets
  - composer.models.gpt2
    Toggle child pages in navigation
    - composer.models.gpt2.gpt2_hparams
    - composer.models.gpt2.model
  - composer.models.initializers
  - composer.models.loss
  - composer.models.model_hparams
  - composer.models.nlp_metrics
  - composer.models.resnet
    Toggle child pages in navigation
    - composer.models.resnet.model
    - composer.models.resnet.resnet_hparams
  - composer.models.resnet20_cifar10
    Toggle child pages in navigation
    - composer.models.resnet20_cifar10.model
    - composer.models.resnet20_cifar10.resnet20_cifar10_hparams
  - composer.models.resnet56_cifar10
    Toggle child pages in navigation
    - composer.models.resnet56_cifar10.model
    - composer.models.resnet56_cifar10.resnet56_cifar10_hparams
  - composer.models.resnet9_cifar10
    Toggle child pages in navigation
    - composer.models.resnet9_cifar10.model
    - composer.models.resnet9_cifar10.resnet9_cifar10_hparams
  - composer.models.resnets
  - composer.models.timm
    Toggle child pages in navigation
    - composer.models.timm.model
    - composer.models.timm.timm_hparams
  - composer.models.transformer_hparams
  - composer.models.transformer_shared
  - composer.models.unet
    Toggle child pages in navigation
- composer.optim
  Toggle child pages in navigation
- composer.profiler
  Toggle child pages in navigation
- composer.trainer
  Toggle child pages in navigation
  - composer.trainer.ddp
  - composer.trainer.devices
    Toggle child pages in navigation
  - composer.trainer.trainer
  - composer.trainer.trainer_hparams
- composer.utils
  Toggle child pages in navigation

v: v0.4.0

Versions: latest; stable; v0.4.0; v0.3.1; v0.3.0; v0.2.4; v0.2.3

Downloads

On Read the Docs: Project Home; Builds

Back to top

Toggle table of contents sidebar

🏎️ Performance#

Under construction 🚧

Made with Sphinx and @pradyunsg's Furo