{
  "model_type": "particlevit",
  "architectures": [
    "ParticleViT"
  ],
  "num_features": 9,
  "num_classes": 210,
  "embed_dim": 1024,
  "depth": 10,
  "num_heads": 16,
  "mlp_ratio": 2.6666666666666665,
  "head_dim": 64,
  "max_particles": 150,
  "num_params": 126091730,
  "pretraining_compute_flops": "1.1e20",
  "auto_map": {
    "AutoModel": "modeling_particlevit.ParticleViT"
  }
}