| { | |
| "architecture": "vit_base_patch16_224", | |
| "num_classes": 0, | |
| "num_features": 768, | |
| "global_pool": "token", | |
| "model_args": { | |
| "img_size": [ | |
| 224, | |
| 224 | |
| ], | |
| "patch_size": 16, | |
| "in_chans": 3, | |
| "num_classes": 0, | |
| "global_pool": "token", | |
| "embed_dim": 768, | |
| "depth": 12, | |
| "num_heads": 12, | |
| "mlp_ratio": 4.0, | |
| "qkv_bias": true, | |
| "qk_norm": false, | |
| "pre_norm": false, | |
| "class_token": true, | |
| "no_embed_class": false, | |
| "reg_tokens": 0 | |
| }, | |
| "pretrained_cfg": { | |
| "architecture": "vit_base_patch16_224", | |
| "input_size": [ | |
| 224, | |
| 224, | |
| 3 | |
| ], | |
| "interpolation": "bicubic", | |
| "crop_pct": 0.9, | |
| "crop_mode": "center", | |
| "mean": [ | |
| 0.485, | |
| 0.456, | |
| 0.406 | |
| ], | |
| "std": [ | |
| 0.229, | |
| 0.224, | |
| 0.225 | |
| ], | |
| "num_classes": 0, | |
| "first_conv": "patch_embed.proj", | |
| "classifier": "head", | |
| "fixed_input_size": true, | |
| "tag": "mae", | |
| "origin_url": "", | |
| "custom_load": false, | |
| "pool_size": null, | |
| "rngs": 42 | |
| } | |
| } |