config.json

{
  "architecture": "ViT (from scratch, custom nn.Module)",
  "img_size": 32,
  "patch_size": 4,
  "in_channels": 3,
  "num_classes": 10,
  "embed_dim": 256,
  "depth": 4,
  "num_heads": 8,
  "mlp_ratio": 4.0,
  "dropout": 0.1,
  "classes": [
    "airplane",
    "automobile",
    "bird",
    "cat",
    "deer",
    "dog",
    "frog",
    "horse",
    "ship",
    "truck"
  ],
  "normalize_mean": [
    0.4914,
    0.4822,
    0.4465
  ],
  "normalize_std": [
    0.247,
    0.2435,
    0.2616
  ],
  "final_test_acc": 77.13
}