Getting Started

Installation
- Docker
- Verification
Using Composer
- Functional API
- MosaicML Trainer
Welcome Tour
Custom Models and Datasets
Distributed Training (DDP)
- Single-Node Example
- Detailed Usage

core

composer.Algorithm
composer.Callback
composer.Engine
- Trace
composer.Event
- Events List
- API Reference
composer.Logger
composer.State
composer.core.surgery
composer.types

composer

composer.algorithms
composer.callbacks
- Callbacks
- Callback Hyperparameters
composer.datasets
- Base Classes and Hyperparameters
- Datasets
composer.functional
composer.loggers
- Backends
- Backend Hyperparameters
composer.models
composer.optim
- DecoupledSGDW
- DecoupledAdamW
composer.optim.scheduler
composer.trainer
composer.trainer.devices
- Devices
- Device Dataloaders

Methods Library

ALiBi
AugMix
BlurPool
Channels Last
ColOut
Cutout
Decoupled Weight Decay
Ghost BatchNorm
Label Smoothing
Layer Freezing
MixUp
Progressive Image Resizing
RandAugment
Scale Schedule
Scaling Laws
Selective Backprop
Sequence Length Warmup
Sharpness Aware Minimization
Squeeze-and-Excitation
Stochastic Depth (Block-Wise)
Stochastic Depth (Sample-Wise)
Stochastic Weight Averaging

Model Library

CIFAR ResNet
EfficientNet
GPT-2
ImageNet ResNet
UNet

MosaicML

»
composer.trainer.devices
Edit on GitHub

composer.trainer.devices

Devices

Device Dataloaders

Previous Next

© Copyright 2021, MosaicML, Inc.. Revision 08944ed6.

Built with Sphinx using a theme provided by Read the Docs.

Read the Docs v: v0.3.0

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

Downloads

On Read the Docs: Project Home; Builds