composer.trainer#

Modules

composer.trainer.ddp

Helpers for running distributed data parallel training.

composer.trainer.devices

Module for devices on which models run.

composer.trainer.trainer

Train models!

composer.trainer.trainer_hparams

The Hparams used to construct the Trainer.

Train models with flexible insertion of algorithms.

Classes

Trainer

Trainer for training a models with Composer algorithms.

Hparams

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

TrainerHparams

Params for instantiating the Trainer.