composer.models.tasks

Modules

composer.models.tasks.classification

A convenience class that creates a ComposerModel for classification tasks from a vanilla PyTorch model.

Model tasks are ComposerModels with forward passes and logging built-in for many common deep learning tasks.

Classes

ComposerClassifier

A convenience class that creates a ComposerModel for classification tasks from a vanilla PyTorch model.