File size: 495 Bytes
c164b10 6e8e247 564d7f2 c164b10 564d7f2 645f152 564d7f2 6e8e247 564d7f2 c164b10 564d7f2 c164b10 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
{
"architectures": ["VisionTransformer"],
"model_type": "vit",
"auto_map": {
"AutoConfig": "vision_transformer.VisionTransformerConfig",
"AutoModel": "vision_transformer.VisionTransformer"
},
"img_size": 224,
"patch_size": 16,
"in_chans": 1,
"num_classes": 0,
"embed_dim": 384,
"depth": 12,
"num_heads": 6,
"mlp_ratio": 4.0,
"qkv_bias": true,
"qk_scale": null,
"drop_rate": 0.0,
"attn_drop_rate": 0.0,
"drop_path_rate": 0.0,
"torch_dtype": "float32"
} |