Getting Started
Installation
Docker
Verification
Using Composer
Functional API
MosaicML Trainer
Welcome Tour
Our First Method!
A Simple Instrumented Trainer
Introducing… Events, Engines, and State
Next: The MosaicML Trainer
Custom Models and Datasets
Models
Datasets
Trainer init
Trainer with YAHP
Distributed Training (DDP)
Single-Node Example
Detailed Usage
core
composer.Algorithm
composer.Callback
Callback
RankZeroCallback
composer.Engine
Trace
composer.Event
Events List
API Reference
composer.Logger
BaseLoggerBackendHparams
BaseLoggerBackend
RankZeroLoggerBackend
composer.State
composer.core.surgery
composer.types
Tensor Types
Batch Types
Dataset and Data Loader Types
Trainer Types
Miscellaneous Types
composer
composer.algorithms
Alibi
Augmix
BlurPool
Channels Last
ColOut
CutOut
Ghost Batch Normalization
Label Smoothing
Layer Freezing
MixUp
Progressive Resizing
RandAugment
Sequence Length Warmup
Sharpness-Aware Minimization
Scaling the Learning Rate Schedule
Selective Backpropagation
Squeeze-and-Excitation
Stochastic Depth
Stochastic Weight Averaging
composer.callbacks
Callbacks
Callback Hyperparameters
composer.datasets
Base Classes and Hyperparameters
Datasets
composer.functional
composer.algorithms.functional.augment_and_mix
composer.algorithms.functional.apply_blurpool
composer.algorithms.functional.apply_alibi
composer.algorithms.functional.colout
composer.algorithms.functional.cutout
composer.algorithms.functional.smooth_labels
composer.algorithms.functional.freeze_layers
composer.algorithms.functional.apply_ghost_batchnorm
composer.algorithms.functional.mixup_batch
composer.algorithms.functional.resize_inputs
composer.algorithms.functional.randaugment
composer.algorithms.functional.scale_scheduler
composer.algorithms.functional.selective_backprop
composer.algorithms.functional.apply_se
composer.algorithms.functional.apply_seq_length_warmup
composer.algorithms.functional.apply_stochastic_depth
composer.loggers
Backends
Backend Hyperparameters
composer.models
Base Models
Image Models
Language Models
Metrics and Loss Functions
composer.optim
DecoupledSGDW
DecoupledAdamW
composer.optim.scheduler
WarmUpLR
ConstantLR
ComposedScheduler
composer.trainer
Examples
Trainer Hparams
API Reference
composer.trainer.devices
Devices
Device Dataloaders
Methods Library
ALiBi
TL;DR
Attribution
Applicable Settings
Hyperparameters
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
AugMix
TL;DR
Attribution
Applicable Settings
Hyperparameters
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
BlurPool
TL;DR
Attribution
Code and Hyperparameters
Applicable Settings
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Acknowledgments
Code
Channels Last
TL;DR
Attribution
Applicable Settings
Hyperparameters
Example Effects
Implementation Details
Considerations
Composability
Code
ColOut
TL;DR
Attribution
Applicable Settings
Hyperparameters
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
Cutout
TL;DR
Attribution
Applicable Settings
Hyperparameters
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
Decoupled Weight Decay
TL;DR
Attribution
Code and Hyperparameters
Applicable Settings
Implementation Details
Considerations
Composability
Ghost BatchNorm
TL;DR
Attribution
Hyperparameters
Applicable Settings
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
Label Smoothing
TL;DR
Attribution
Applicable Settings
Hyperparameters
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
Layer Freezing
TL;DR
Attribution
Hyperparameters
Applicable Settings
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
MixUp
TL;DR
Attribution
Hyperparameters
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
Progressive Image Resizing
TL;DR
Attribution
Applicable Settings
Hyperparameters
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Detailed Results
Code
RandAugment
TL;DR
Attribution
Applicable Settings
Hyperparameters
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
Scale Schedule
TL;DR
Attribution
Hyperparameters
Example Effects
Implementation Details
Suggested Hyperparameters
Composability
Code
Scaling Laws
TL;DR
Attribution
Applicable Settings
Hyperparameters
Detailed Results
Considerations
Effects & Implications
Selective Backprop
TL;DR
Attribution
Applicable Settings
Hyperparameters
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Detailed Results
Code
Sequence Length Warmup
TL;DR
Hyperparameters
Applicable Settings
Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composition
Attribution
Code
Sharpness Aware Minimization
TL;DR
Attribution
Code and Hyperparameters
Applicable Settings
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
Squeeze-and-Excitation
TL;DR
Attribution
Code and Hyperparameters
Applicable Settings
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
Stochastic Depth (Block-Wise)
TL;DR
Attribution
Applicable Settings
Hyperparameters
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
Stochastic Depth (Sample-Wise)
TL;DR
Attribution
Hyperparameters
Applicable Settings
Example Effects
Implementation Details
Suggested Hyperparameters
Considerations
Composability
Code
Stochastic Weight Averaging
TL;DR
Attribution
Applicable Settings
Hyperparameters
Example Effects
Implementation Details
Considerations
Composability
Model Library
CIFAR ResNet
Overview
Attribution
Architecture
Family members
Default Training Hyperparameters
EfficientNet
Overview
Attribution
Architecture
Family members
Default Training Hyperparameters
GPT-2
Overview
Attribution
Architecture
Family Members
Implementation Details
Exploring Tradeoffs Between Quality and Training Speed / Cost
ImageNet ResNet
Overview
Attribution
Architecture
Family Members
Implementation details
Default Training Hyperparameters
UNet
Overview
Attribution
Architecture
Implementation Details
Exploring Tradeoffs Between Quality and Training Speed/Cost
MosaicML
»
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
I
|
J
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
_
_log_metric() (composer.core.logging.base_backend.RankZeroLoggerBackend method)
_training_start() (composer.core.logging.base_backend.RankZeroLoggerBackend method)
_will_log() (composer.core.logging.base_backend.RankZeroLoggerBackend method)
A
AFTER_BACKWARD (composer.Event attribute)
after_backward() (composer.Callback method)
(composer.loggers.tqdm_logger.TQDMLoggerBackend method)
AFTER_DATALOADER (composer.Event attribute)
after_dataloader() (composer.Callback method)
AFTER_FORWARD (composer.Event attribute)
after_forward() (composer.Callback method)
AFTER_LOSS (composer.Event attribute)
after_loss() (composer.Callback method)
AFTER_TRAIN_BATCH (composer.Event attribute)
after_train_batch() (composer.Callback method)
(composer.callbacks.grad_monitor.GradMonitor method)
Algorithm (class in composer)
algorithms (composer.State attribute)
Alibi (class in composer.algorithms.alibi)
AlibiHparams (class in composer.algorithms.alibi)
AMP (composer.types.Precision attribute)
apply() (composer.Algorithm method)
(composer.algorithms.augmix.AugMix method)
(composer.algorithms.seq_length_warmup.SeqLengthWarmup method)
apply_alibi() (in module composer.algorithms.alibi)
(in module composer.algorithms.functional)
apply_blurpool() (in module composer.algorithms.blurpool)
(in module composer.algorithms.functional)
apply_ghost_batchnorm() (in module composer.algorithms.functional)
(in module composer.algorithms.ghost_batchnorm)
apply_se() (in module composer.algorithms.functional)
(in module composer.algorithms.squeeze_excite)
apply_seq_length_warmup (class in composer.algorithms.seq_length_warmup)
apply_seq_length_warmup() (in module composer.algorithms.functional)
(in module composer.algorithms.seq_length_warmup)
apply_stochastic_depth() (in module composer.algorithms.functional)
(in module composer.algorithms.stochastic_depth)
as_batch_dict() (in module composer.types)
as_batch_pair() (in module composer.types)
augment_and_mix() (in module composer.algorithms.augmix)
(in module composer.algorithms.functional)
AugMix (class in composer.algorithms.augmix)
,
[1]
AugMixHparams (class in composer.algorithms.augmix)
B
backends (composer.Logger attribute)
backwards_create_graph (composer.Algorithm property)
BaseLoggerBackend (class in composer.core.logging.base_backend)
BaseLoggerBackendHparams (class in composer.loggers.logger_hparams)
BaseMosaicModel (class in composer.models)
Batch (class in composer.types)
batch (composer.State attribute)
batch_dict (composer.State property)
BATCH_END (composer.Event attribute)
batch_end() (composer.Callback method)
(composer.callbacks.benchmarker.Benchmarker method)
(composer.callbacks.lr_monitor.LRMonitor method)
(composer.callbacks.speed_monitor.SpeedMonitor method)
(composer.callbacks.torch_profiler.TorchProfiler method)
(composer.loggers.file_logger.FileLoggerBackend method)
(composer.loggers.wandb_logger.WandBLoggerBackend method)
batch_idx (composer.State property)
batch_pair (composer.State property)
batch_size (composer.types.DataLoader attribute)
BATCH_START (composer.Event attribute)
batch_start() (composer.Callback method)
(composer.callbacks.benchmarker.Benchmarker method)
(composer.callbacks.speed_monitor.SpeedMonitor method)
(composer.callbacks.torch_profiler.TorchProfiler method)
BatchDict (class in composer.types)
BatchPair (class in composer.types)
BEFORE_BACKWARD (composer.Event attribute)
before_backward() (composer.Callback method)
BEFORE_FORWARD (composer.Event attribute)
before_forward() (composer.Callback method)
BEFORE_LOSS (composer.Event attribute)
before_loss() (composer.Callback method)
BEFORE_TRAIN_BATCH (composer.Event attribute)
before_train_batch() (composer.Callback method)
Benchmarker (class in composer.callbacks.benchmarker)
BenchmarkerHparams (class in composer.callbacks.callback_hparams)
blur_2d() (in module composer.algorithms.blurpool)
BlurConv2d (class in composer.algorithms.blurpool)
BlurMaxPool2d (class in composer.algorithms.blurpool)
BlurPool (class in composer.algorithms.blurpool)
BlurPool2d (class in composer.algorithms.blurpool)
BlurPoolHparams (class in composer.algorithms.blurpool)
BratsDatasetHparams (class in composer.datasets)
BreakEpochException
C
Callback (class in composer)
callbacks (composer.State attribute)
ChannelsLast (class in composer.algorithms.channels_last)
ChannelsLastHparams (class in composer.algorithms.channels_last)
CIFAR10_ResNet56 (class in composer.models)
CIFAR10DatasetHparams (class in composer.datasets)
collate_fn (composer.datasets.DataloaderSpec attribute)
,
[1]
ColOut (class in composer.algorithms.colout)
colout() (in module composer.algorithms.colout)
(in module composer.algorithms.functional)
ColOutHparams (class in composer.algorithms.colout)
ComposedScheduler (class in composer.optim)
composer.algorithms.seq_length_warmup
module
composer.core.surgery
module
compute() (composer.models.loss.CrossEntropyLoss method)
(composer.models.loss.Dice method)
(composer.models.nlp_metrics.LanguageCrossEntropyLoss method)
(composer.models.nlp_metrics.Perplexity method)
ConstantLR (class in composer.optim)
count_module_instances() (in module composer.core.surgery)
create_from_hparams() (composer.Trainer class method)
CrossEntropyLoss (class in composer.models.loss)
CudaDataLoader (class in composer.trainer.devices.device_gpu)
CutOut (class in composer.algorithms.cutout)
cutout() (in module composer.algorithms.cutout)
(in module composer.algorithms.functional)
CutOutHparams (class in composer.algorithms.cutout)
D
DataLoader (class in composer.types)
dataloader_to_device() (composer.trainer.devices.device.Device method)
(composer.trainer.devices.device_cpu.DeviceCPU method)
(composer.trainer.devices.device_gpu.DeviceGPU method)
DataloaderHparams (class in composer.datasets)
DataloaderSpec (class in composer.datasets)
Dataset (class in composer.types)
dataset (composer.datasets.DataloaderSpec attribute)
,
[1]
(composer.types.DataLoader attribute)
DatasetHparams (class in composer.datasets)
ddp_backend (composer.trainer.devices.device.Device property)
(composer.trainer.devices.device_cpu.DeviceCPU property)
(composer.trainer.devices.device_gpu.DeviceGPU property)
DecoupledAdamW (class in composer.optim)
DecoupledSGDW (class in composer.optim)
Device (class in composer.trainer.devices.device)
DeviceCPU (class in composer.trainer.devices.device_cpu)
DeviceGPU (class in composer.trainer.devices.device_gpu)
Dice (class in composer.models.loss)
drop_last (composer.datasets.DataloaderSpec attribute)
,
[1]
(composer.types.DataLoader attribute)
E
EfficientNetB0 (class in composer.models)
Engine (class in composer)
engine (composer.Trainer attribute)
epoch (composer.State attribute)
EPOCH_END (composer.Event attribute)
epoch_end() (composer.Callback method)
(composer.callbacks.benchmarker.Benchmarker method)
(composer.callbacks.speed_monitor.SpeedMonitor method)
(composer.loggers.tqdm_logger.TQDMLoggerBackend method)
(composer.loggers.wandb_logger.WandBLoggerBackend method)
EPOCH_START (composer.Event attribute)
epoch_start() (composer.Callback method)
(composer.callbacks.speed_monitor.SpeedMonitor method)
(composer.callbacks.torch_profiler.TorchProfiler method)
(composer.loggers.tqdm_logger.TQDMLoggerBackend method)
eval() (composer.Trainer method)
EVAL_AFTER_FORWARD (composer.Event attribute)
eval_after_forward() (composer.Callback method)
(composer.loggers.tqdm_logger.TQDMLoggerBackend method)
EVAL_BATCH_END (composer.Event attribute)
eval_batch_end() (composer.Callback method)
eval_batch_size (composer.State attribute)
EVAL_BATCH_START (composer.Event attribute)
eval_batch_start() (composer.Callback method)
EVAL_BEFORE_FORWARD (composer.Event attribute)
eval_before_forward() (composer.Callback method)
eval_dataloader (composer.State attribute)
EVAL_END (composer.Event attribute)
eval_end() (composer.Callback method)
(composer.loggers.tqdm_logger.TQDMLoggerBackend method)
EVAL_START (composer.Event attribute)
eval_start() (composer.Callback method)
(composer.loggers.tqdm_logger.TQDMLoggerBackend method)
Event (class in composer)
exit_code (composer.core.engine.Trace attribute)
F
FileLoggerBackend (class in composer.loggers.file_logger)
FileLoggerBackendHparams (class in composer.loggers.logger_hparams)
find_unused_parameters (composer.Algorithm property)
fit() (composer.Trainer method)
forward() (composer.models.BaseMosaicModel method)
(composer.models.MosaicClassifier method)
(composer.models.MosaicTransformer method)
(composer.models.UNet method)
FP32 (composer.types.Precision attribute)
freeze_layers() (in module composer.algorithms.functional)
(in module composer.algorithms.layer_freezing)
G
generator (composer.datasets.DataloaderSpec attribute)
,
[1]
get_model_inputs() (composer.models.MosaicTransformer method)
GhostBatchNorm (class in composer.algorithms.ghost_batchnorm)
GhostBatchNormHparams (class in composer.algorithms.ghost_batchnorm)
GPT2Model (class in composer.models)
grad_accum (composer.State attribute)
GradMonitor (class in composer.callbacks.grad_monitor)
GradMonitorHparams (class in composer.callbacks.callback_hparams)
I
ImagenetDatasetHparams (class in composer.datasets)
inference2d() (composer.models.UNet method)
INIT (composer.Event attribute)
init() (composer.Callback method)
initialize_object() (composer.callbacks.callback_hparams.BenchmarkerHparams method)
(composer.callbacks.callback_hparams.GradMonitorHparams method)
(composer.callbacks.callback_hparams.LRMonitorHparams method)
(composer.callbacks.callback_hparams.SpeedMonitorHparams method)
(composer.callbacks.callback_hparams.TorchProfilerHparams method)
(composer.datasets.BratsDatasetHparams method)
(composer.datasets.CIFAR10DatasetHparams method)
(composer.datasets.DataloaderHparams method)
(composer.datasets.DatasetHparams method)
(composer.datasets.ImagenetDatasetHparams method)
(composer.datasets.LMDatasetHparams method)
(composer.datasets.MNISTDatasetHparams method)
(composer.datasets.SyntheticDatasetHparams method)
(composer.loggers.logger_hparams.BaseLoggerBackendHparams method)
(composer.loggers.logger_hparams.FileLoggerBackendHparams method)
(composer.loggers.logger_hparams.TQDMLoggerBackendHparams method)
(composer.loggers.logger_hparams.WandBLoggerBackendHparams method)
J
JSON (class in composer.types)
L
LabelSmoothing (class in composer.algorithms.label_smoothing)
LabelSmoothingHparams (class in composer.algorithms.label_smoothing)
LanguageCrossEntropyLoss (class in composer.models.nlp_metrics)
last_batch_size (composer.State attribute)
LayerFreezing (class in composer.algorithms.layer_freezing)
LayerFreezingHparams (class in composer.algorithms.layer_freezing)
LMDatasetHparams (class in composer.datasets)
load_state_dict() (composer.callbacks.speed_monitor.SpeedMonitor method)
(composer.callbacks.torch_profiler.TorchProfiler method)
(composer.loggers.tqdm_logger.TQDMLoggerBackend method)
(composer.State method)
(composer.trainer.devices.device_cpu.DeviceCPU method)
(composer.trainer.devices.device_gpu.DeviceGPU method)
(composer.types.Serializable method)
log_metric() (composer.core.logging.base_backend.BaseLoggerBackend method)
(composer.core.logging.base_backend.RankZeroLoggerBackend method)
Logger (class in composer)
logger (composer.Trainer attribute)
loss (composer.State attribute)
loss() (composer.models.BaseMosaicModel method)
(composer.models.GPT2Model method)
(composer.models.MosaicClassifier method)
(composer.models.MosaicTransformer method)
(composer.models.UNet method)
LRMonitor (class in composer.callbacks.lr_monitor)
LRMonitorHparams (class in composer.callbacks.callback_hparams)
M
match() (composer.Algorithm method)
(composer.algorithms.augmix.AugMix method)
(composer.algorithms.seq_length_warmup.SeqLengthWarmup method)
max_epochs (composer.State attribute)
Metrics (class in composer.types)
metrics() (composer.models.BaseMosaicModel method)
(composer.models.GPT2Model method)
(composer.models.MosaicClassifier method)
(composer.models.MosaicTransformer method)
(composer.models.UNet method)
MixUp (class in composer.algorithms.mixup)
mixup_batch() (in module composer.algorithms.functional)
(in module composer.algorithms.mixup)
MixUpHparams (class in composer.algorithms.mixup)
MNIST_Classifier (class in composer.models)
MNISTDatasetHparams (class in composer.datasets)
Model (class in composer.types)
model (composer.State attribute)
ModelParameters (class in composer.types)
module
composer.algorithms.seq_length_warmup
composer.core.surgery
module_to_device() (composer.trainer.devices.device.Device method)
(composer.trainer.devices.device_cpu.DeviceCPU method)
(composer.trainer.devices.device_gpu.DeviceGPU method)
MosaicClassifier (class in composer.models)
MosaicTransformer (class in composer.models)
move_to_gpu() (composer.trainer.devices.device_gpu.CudaDataLoader method)
multiprocessing_context (composer.datasets.DataloaderSpec attribute)
,
[1]
N
normal_iter() (composer.trainer.devices.device_gpu.CudaDataLoader method)
num_workers (composer.types.DataLoader attribute)
O
Optimizer (class in composer.types)
optimizer_to_device() (composer.trainer.devices.device.Device method)
Optimizers (class in composer.types)
optimizers (composer.State attribute)
order (composer.core.engine.Trace attribute)
outputs (composer.State attribute)
P
Perplexity (class in composer.models.nlp_metrics)
pin_memory (composer.types.DataLoader attribute)
Precision (class in composer.types)
precision (composer.State attribute)
precision_context() (composer.trainer.devices.device.Device method)
(composer.trainer.devices.device_cpu.DeviceCPU method)
(composer.trainer.devices.device_gpu.DeviceGPU method)
prefetch_factor (composer.types.DataLoader attribute)
prefetch_fn (composer.datasets.DataloaderSpec attribute)
,
[1]
prefetch_in_cuda_stream_iter() (composer.trainer.devices.device_gpu.CudaDataLoader method)
prepare() (composer.trainer.devices.device.Device method)
(composer.trainer.devices.device_cpu.DeviceCPU method)
(composer.trainer.devices.device_gpu.DeviceGPU method)
ProgressiveResizing (class in composer.algorithms.progressive_resizing)
ProgressiveResizingHparams (class in composer.algorithms.progressive_resizing)
R
RandAugment (class in composer.algorithms.randaugment)
randaugment() (in module composer.algorithms.functional)
(in module composer.algorithms.randaugment)
RandAugmentHparams (class in composer.algorithms.randaugment)
RankZeroCallback (class in composer.core.callback)
RankZeroLoggerBackend (class in composer.core.logging.base_backend)
replace_module_classes() (in module composer.core.surgery)
ReplacementFunction (class in composer.core.surgery)
resize_inputs() (in module composer.algorithms.functional)
(in module composer.algorithms.progressive_resizing)
ResNet101 (class in composer.models)
ResNet18 (class in composer.models)
ResNet50 (class in composer.models)
run (composer.core.engine.Trace attribute)
run_event() (composer.Engine method)
S
SAM (class in composer.algorithms.sam)
SAMHparams (class in composer.algorithms.sam)
sampler (composer.types.DataLoader attribute)
scale_scheduler() (in module composer.algorithms.functional)
(in module composer.algorithms.scale_schedule)
Scaler (class in composer.types)
scaler (composer.State attribute)
ScaleSchedule (class in composer.algorithms.scale_schedule)
ScaleScheduleHparams (class in composer.algorithms.scale_schedule)
Scheduler (class in composer.types)
Schedulers (class in composer.types)
schedulers (composer.State attribute)
selective_backprop() (in module composer.algorithms.functional)
SelectiveBackprop (class in composer.algorithms.selective_backprop)
SelectiveBackpropHparams (class in composer.algorithms.selective_backprop)
SeqLengthWarmup (class in composer.algorithms.seq_length_warmup)
,
[1]
SeqLengthWarmupHparams (class in composer.algorithms.seq_length_warmup)
Serializable (class in composer.types)
shuffle (composer.datasets.DataloaderSpec attribute)
,
[1]
smooth_labels() (in module composer.algorithms.functional)
(in module composer.algorithms.label_smoothing)
soft_cross_entropy() (in module composer.models.loss)
SpeedMonitor (class in composer.callbacks.speed_monitor)
SpeedMonitorHparams (class in composer.callbacks.callback_hparams)
split_fn (composer.datasets.DataloaderSpec attribute)
,
[1]
SqueezeExcite (class in composer.algorithms.squeeze_excite)
SqueezeExcite2d (class in composer.algorithms.squeeze_excite)
SqueezeExciteConv2d (class in composer.algorithms.squeeze_excite)
SqueezeExciteHparams (class in composer.algorithms.squeeze_excite)
State (class in composer)
state (composer.Trainer attribute)
state_dict() (composer.callbacks.speed_monitor.SpeedMonitor method)
(composer.callbacks.torch_profiler.TorchProfiler method)
(composer.loggers.tqdm_logger.TQDMLoggerBackend method)
(composer.loggers.wandb_logger.WandBLoggerBackend method)
(composer.State method)
(composer.trainer.devices.device_cpu.DeviceCPU method)
(composer.trainer.devices.device_gpu.DeviceGPU method)
(composer.types.Serializable method)
StateDict (class in composer.types)
step (composer.State attribute)
steps_per_epoch (composer.State property)
StochasticBottleneck (class in composer.algorithms.stochastic_depth)
StochasticDepth (class in composer.algorithms.stochastic_depth)
StochasticDepthHparams (class in composer.algorithms.stochastic_depth)
SWA (class in composer.algorithms.swa)
SWAHparams (class in composer.algorithms.swa)
SyntheticDatasetHparams (class in composer.datasets)
T
Tensor (class in composer.types)
tensor_to_device() (composer.trainer.devices.device.Device method)
(composer.trainer.devices.device_cpu.DeviceCPU method)
(composer.trainer.devices.device_gpu.DeviceGPU method)
Tensors (class in composer.types)
timeout (composer.types.DataLoader attribute)
TorchProfiler (class in composer.callbacks.torch_profiler)
TorchProfilerHparams (class in composer.callbacks.callback_hparams)
TPrefetchFn (class in composer.types)
TQDMLoggerBackend (class in composer.loggers.tqdm_logger)
TQDMLoggerBackendHparams (class in composer.loggers.logger_hparams)
Trace (class in composer.core.engine)
train_batch_size (composer.State attribute)
train_dataloader (composer.State attribute)
Trainer (class in composer)
TRAINING_END (composer.Event attribute)
training_end() (composer.Callback method)
TRAINING_START (composer.Event attribute)
training_start() (composer.Callback method)
(composer.callbacks.benchmarker.Benchmarker method)
(composer.callbacks.torch_profiler.TorchProfiler method)
(composer.core.logging.base_backend.RankZeroLoggerBackend method)
U
UNet (class in composer.models)
update() (composer.models.loss.CrossEntropyLoss method)
(composer.models.loss.Dice method)
(composer.models.nlp_metrics.LanguageCrossEntropyLoss method)
V
validate() (composer.datasets.SyntheticDatasetHparams method)
(composer.models.BaseMosaicModel method)
(composer.models.MosaicClassifier method)
(composer.models.MosaicTransformer method)
(composer.models.UNet method)
W
WandBLoggerBackend (class in composer.loggers.wandb_logger)
WandBLoggerBackendHparams (class in composer.loggers.logger_hparams)
WarmUpLR (class in composer.optim)
will_log() (composer.core.logging.base_backend.BaseLoggerBackend method)
(composer.core.logging.base_backend.RankZeroLoggerBackend method)
worker_init_fn (composer.datasets.DataloaderSpec attribute)
,
[1]
Read the Docs
v: v0.3.1
Versions
latest
stable
v0.3.1
v0.3.0
v0.2.4
v0.2.3
Downloads
On Read the Docs
Project Home
Builds