| { | |
| "architecture": "vit_base_patch16_224", | |
| "num_classes": 527, | |
| "num_features": 768, | |
| "global_pool": "avg", | |
| "pretrained_cfg": { | |
| "tag": "audiomae", | |
| "custom_load": false, | |
| "input_size": [ | |
| 1, | |
| 1024, | |
| 128 | |
| ], | |
| "fixed_input_size": true, | |
| "num_classes": 527, | |
| "pool_size": null, | |
| "first_conv": "patch_embed.proj", | |
| "classifier": "head", | |
| "license": "cc-by-nc-4.0" | |
| }, | |
| "model_args": { | |
| "global_pool": "avg" | |
| } | |
| } |