config.json

{"image_size": 32, "in_channels": 3, "num_classes": 7, "emb_dim": 128, "num_heads": 4, "depth": 6, "mlp_ratio": 4.0, "drop": 0.1, "patch_size": 4, "batch_size": 8, "epochs": 20}