| { | |
| "model_type": "particlevit", | |
| "architectures": [ | |
| "ParticleViT" | |
| ], | |
| "num_features": 9, | |
| "num_classes": 210, | |
| "embed_dim": 512, | |
| "depth": 5, | |
| "num_heads": 8, | |
| "mlp_ratio": 2.6666666666666665, | |
| "head_dim": 64, | |
| "max_particles": 150, | |
| "num_params": 15857490, | |
| "pretraining_compute_flops": "1.4e19", | |
| "auto_map": { | |
| "AutoModel": "modeling_particlevit.ParticleViT" | |
| } | |
| } |