composer.models

Models provided to Trainer must use the basic interface specified by BaseMosaicModel.

Additionally, for convience we provide a number of extensions of BaseMosaicModel as detailed below.

Base Models

Image Models

Language Models

Metrics and Loss Functions

Evaluation metrics for common tasks are in torchmetrics and are directly compatible with BaseMosaicModel. Additionally, we provide implementations of the following metrics and loss functions.