class composer.models.ViTSmallPatch16Hparams(initializers=<factory>, num_classes=1000, image_size=224, channels=3, dropout=0.0, embedding_dropout=0.0)[source]#

YAHP interface for vit_small_batch16.

  • num_classes (int, optional) โ€“ number of classes for the model. Default: 1000.

  • image_size (int, optional) โ€“ input image size. If you have rectangular images, make sure your image size is the maximum of the width and height. Default: 224.

  • channels (int, optional) โ€“ number of image channels. Default: 3.

  • dropout (float, optional) โ€“ 0.0 - 1.0 dropout rate. Default: 0.

  • embedding_dropout (float, optional) โ€“ 0.0 - 1.0 embedding dropout rate. Default: 0.