vit_b_16

torchvision.models.vit_b_16(pretrained: bool = False, progress: bool = True, **kwargs: Any) → torchvision.models.vision_transformer.VisionTransformer

Constructs a vit_b_16 architecture from “An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale”.

Parameters

• pretrained (bool) – If True, returns a model pre-trained on ImageNet

• progress (bool) – If True, displays a progress bar of the download to stderr