towhee.models.vit.vit

Functions

compute_rollout_attention
create_model	Create ViT model.

Classes

VitModel

Vision Transformer Model :param img_size: Image height or width (height=width).