config.json

{
  "image_size": 224,
  "patch_size": 28,
  "num_classes": 1001,
  "dim": 384,
  "depth": 6,
  "heads": 6,
  "mlp_dim": 1472,
  "pool": "cls",
  "channels": 3,
  "dim_head": 64,
  "dropout": 0.1,
  "emb_dropout": 0.05
}