| { | |
| "model_type": "particlevit", | |
| "architectures": [ | |
| "ParticleViT" | |
| ], | |
| "num_features": 9, | |
| "num_classes": 210, | |
| "embed_dim": 1536, | |
| "depth": 14, | |
| "num_heads": 24, | |
| "mlp_ratio": 2.6666666666666665, | |
| "head_dim": 64, | |
| "max_particles": 150, | |
| "num_params": 396810706, | |
| "pretraining_compute_flops": "3.4e20", | |
| "auto_map": { | |
| "AutoModel": "modeling_particlevit.ParticleViT" | |
| } | |
| } |