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

Train models with Composer algorithms.

Hparams

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

EvalHparams

Hyperparameters to describe a call to Trainer.eval().

ExperimentHparams

Hyperparameters to describe an experiment.

FitHparams

Hyperparameters to describe a call to Trainer.fit().

TrainerHparams

Params for instantiating the Trainer.