config.json

{
  "depth": 4,
  "embed_dim": 512,
  "img_size": 256,
  "in_channels": 3,
  "mlp_dim": 1024,
  "num_classes": 1000,
  "num_heads": 16,
  "patch_size": 16
}