{
  "model_type": "particlevit",
  "architectures": [
    "ParticleViT"
  ],
  "num_features": 9,
  "num_classes": 210,
  "embed_dim": 832,
  "depth": 8,
  "num_heads": 13,
  "mlp_ratio": 2.6666666666666665,
  "head_dim": 64,
  "max_particles": 150,
  "num_params": 66663570,
  "pretraining_compute_flops": "5.7e19",
  "auto_map": {
    "AutoModel": "modeling_particlevit.ParticleViT"
  }
}