Files
hydra-ml-config-template/examples/02_ml_training/conf/model/transformer.yaml
2025-10-12 19:39:12 +08:00

8 lines
106 B
YAML

name: ViT-B/16
layers: 12
pretrained: true
num_classes: 1000
hidden_dim: 768
num_heads: 12
patch_size: 16