config.json

{
  "architectures": [
    "SAGViTClassifier"
  ],
  "d_model": 64,
  "dim_feedforward": 64,
  "gcn_hidden": 128,
  "gcn_out": 64,
  "hidden_mlp_features": 64,
  "in_channels": 2560,
  "model_type": "sagvit",
  "nhead": 4,
  "num_classes": 10,
  "num_layers": 2,
  "patch_size": [
    4,
    4
  ],
  "torch_dtype": "float32",
  "transformers_version": "4.47.0"
}