models#

Modules

`composer.models.base`	The ComposerModel base interface.
`composer.models.bert`	The BERT model family using Hugging Face Transformers.
`composer.models.classify_mnist`	A simple example convolutional neural network which can be used to classify MNIST data.
`composer.models.deeplabv3`	DeepLabV3 for image segmentation.
`composer.models.efficientnetb0`	The EfficientNet model family is a set of convolutional neural networks that can be used as the basis for a variety of vision tasks, but were initially designed for image classification.
`composer.models.gpt2`	The GPT-2 model family is set of transformer-based networks for autoregressive language modeling at various scales.
`composer.models.huggingface`	A wrapper class that converts 🤗 Transformers models to composer models
`composer.models.initializers`	Module Initializers.
`composer.models.model_hparams`	General YAHP interface for Base ComposerModels.
`composer.models.resnet`	The ResNet model family is a set of convolutional neural networks described in Deep Residual Learning for Image Recognition (He et al, 2015).
`composer.models.resnet_cifar`	A ResNet model family adapted for CIFAR10 image sizes.
`composer.models.ssd`	SSD300 for object detection on MSCOCO.
`composer.models.tasks`	Model tasks are ComposerModels with forward passes and logging built-in for many common deep learning tasks.
`composer.models.timm`	A wrapper around timm.create_model() used to create `ComposerClassifier`.
`composer.models.unet`	The Unet architecture used in image segmentation.
`composer.models.vit_small_patch16`([...])	Helper function to create a `ComposerClassifier` using a ViT-S/16 model.

The models module contains the ComposerModel base class along with reference implementations of many common models. Additionally, it includes task-specific convenience ComposerModels that wrap existing Pytorch models with standard forward passes and logging to enable quick interaction with the Trainer.

See Composer Model for more details.

Functions

`composer_deeplabv3`	Helper function to create a `ComposerClassifier` with a DeepLabv3(+) model. Logs
`composer_efficientnetb0`	Helper function to create a `ComposerClassifier` with an EfficientNet-b0 architecture.
`composer_resnet`	Helper function to create a `ComposerClassifier` with a torchvision ResNet model.
`composer_resnet_cifar`	Helper function to create a `ComposerClassifier` with a CIFAR ResNet models.
`composer_timm`	A wrapper around timm.create_model() used to create `ComposerClassifier`.
`create_bert_classification`	BERT classification model based on 🤗 Transformers.
`create_bert_mlm`	BERT model based on 🤗 Transformers.
`create_gpt2`	Implements `HuggingFaceModel` to wrap Hugging Face GPT-2 transformers. Logs training and
`mnist_model`	Helper function to create a `ComposerClassifier` with a simple convolutional neural network.
`vit_small_patch16`	Helper function to create a `ComposerClassifier` using a ViT-S/16 model.

Classes

`ComposerClassifier`	A convenience class that creates a `ComposerModel` for classification tasks from a vanilla PyTorch model.
`ComposerModel`	The interface needed to make a PyTorch model compatible with `composer.Trainer`.
`HuggingFaceModel`	A wrapper class that converts 🤗 Transformers models to composer models.
`Initializer`	Sets the initialization scheme for different layers of a PyTorch model.
`SSD`	Single Shot Object detection Model with pretrained ResNet34 backbone extending `ComposerModel`.
`UNet`	A U-Net model extending `ComposerModel`.

Hparams

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

`BERTForClassificationHparams`	YAHP interface for `BERTModel`.
`BERTHparams`	YAHP interface for `BERTModel`.
`DeepLabV3Hparams`	YAHP interface for
`EfficientNetB0Hparams`	YAHP interface for
`GPT2Hparams`	YAHP interface for `GPT2Model`.
`MnistClassifierHparams`	YAHP interface for `mnist_model()`.
`ModelHparams`	General YAHP interface for ComposerModels.
`ResNetCIFARHparams`	`Hparams` interface for `composer_resnet_cifar()`.
`ResNetHparams`	YAHP interface for `composer_resnet()`.
`SSDHparams`	YAHP interface for `SSD`.
`TimmHparams`	YAHP interface for `composer_timm()`.
`UnetHparams`	YAHP interface for `UNet`.
`ViTSmallPatch16Hparams`	YAHP interface for `vit_small_batch16`.