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