config.json

{
  "model_type": "sag-vit",
  "d_model": 64,
  "dim_feedforward": 64,
  "gcn_hidden": 128,
  "gcn_out": 64,
  "hidden_mlp_features": 64,
  "in_channels": 2560,
  "nhead": 4,
  "num_classes": 10,
  "num_layers": 2,
  "patch_size": [
    4,
    4
  ]
}