config.json

{
  "dropout": 0.2,
  "emb_dim": 512,
  "ff_hidden_dim_multiplier": 4,
  "heads": 16,
  "img_size": 224,
  "num_classes": 4,
  "num_layers": 12,
  "patch_size": 16
}