composer.models.vit_small_patch16

Modules

composer.models.vit_small_patch16.hparams	YAHP interface for ViTSmallPatch16.
composer.models.vit_small_patch16.model	Implements ViT-S/16 as a ComposerClassifier.

ViT Small Patch 16 for image classification.

Classes

ViTSmallPatch16

Implements ViT-S/16 as a ComposerClassifier.

Hparams

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

ViTSmallPatch16Hparams

YAHP interface for ViTSmallPatch16.