config.json

{
  "hidden_size": 128,
  "num_heads": 4,
  "mlp_ratio": 4.0,
  "num_blocks": 4,
  "grid_size": 30,
  "num_classes": 3,
  "schedule": "cosine-high"
}