config.json

{
  "channels": 3,
  "depth": 6,
  "embed_dim": 256,
  "img_size": 32,
  "mlp_dim": 1024,
  "num_classes": 1000,
  "patch_size": 8
}