config.json

{
  "model_type": "particlevit",
  "architectures": [
    "ParticleViT"
  ],
  "num_features": 9,
  "num_classes": 210,
  "embed_dim": 640,
  "depth": 7,
  "num_heads": 10,
  "mlp_ratio": 2.6666666666666665,
  "head_dim": 64,
  "max_particles": 150,
  "num_params": 34566866,
  "pretraining_compute_flops": "3.0e19",
  "auto_map": {
    "AutoModel": "modeling_particlevit.ParticleViT"
  }
}