Skip to content

config

ViTForImageClassificationConfig dataclass

Bases: AnyModelForImageClassificationConfig

Config for ViT model

Source code in glasses/models/vision/image/classification/vit/config.py 
 9
10
11
12
13
14
@dataclass
class ViTForImageClassificationConfig(AnyModelForImageClassificationConfig):
    """Config for [`ViT`](/models/vision/image/classification/vit) model"""

    backbone_config: ViTBackboneConfig
    head_config: ViTHeadConfig