composer.utils

Modules

composer.utils.dist	Helper methods for torch.distributed.
composer.utils.iter_helpers	Utilities for iterating over collections.
composer.utils.module_surgery	Modify model architectures.
composer.utils.object_store	Utility for uploading to and downloading from cloud object stores.
composer.utils.reproducibility	Helper utilities for configuring deterministic training to ensure reproducibility.
composer.utils.run_directory	The run directory stores artifacts created during training.
composer.utils.string_enum	Base class for Enums containing string values.

Helper utilities.

Functions

ensure_tuple	Converts x into a tuple.
iterate_with_pbar	Iterate over a batch iterator and update a tqdm.tqdm progress bar by the batch size on each step.
map_collection	Apply map_fn on each element in collection.

Classes

ObjectStoreProvider	Utility for uploading to and downloading from object (blob) stores, such as Amazon S3.
StringEnum	Base class for Enums containing string values.

Hparams

These classes are used with yahp for YAML-based configuration.

ObjectStoreProviderHparams

ObjectStoreProvider hyperparameters.