{
  "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"
  }
}