{
  "embed_dim": 256,
  "depth": 6,
  "num_heads": 8,
  "mlp_dim": 1024,
  "img_size": 32,
  "patch_size": 4,
  "in_channels": 3,
  "num_classes": 10,
  "dropout": 0.1,
  "pos_embed_type": "1d",
  "aug_type": "augment_best"
}