config.json

{
  "dataset": "cifar10",
  "img": 32,
  "patch": 4,
  "in_ch": 3,
  "dim": 384,
  "depth": 12,
  "heads": 6,
  "num_classes": 10,
  "mlp_ratio": 4,
  "T": 1000,
  "batch": 128,
  "lr": 0.0001,
  "ema_decay": 0.9999,
  "lam_energy": 0.002,
  "tau": 1.0,
  "class_dropout": 0.1,
  "target_keep": 0.5,
  "always_on": 2,
  "grad_clip": 1.0,
  "run_tag": "baseline",
  "total_steps": 200000,
  "warmup_energy_steps": 3000,
  "base_seed": 1234,
  "ckpt_every_steps": 1000,
  "push_interval_min": 20,
  "min_push_gap_min": 5,
  "disk_cap_gb": 17.0,
  "hf_user": "Shanmuk4622",
  "device": "cuda",
  "amp": true,
  "grad_ckpt": false,
  "num_workers": 2,
  "smoke": false,
  "use_head_axis": false,
  "calibrated_cost_repo": null
}