baseline_s2/config.json

{
  "variant": "baseline",
  "seed": 2,
  "out": "C:\\Users\\Bhorn\\Desktop\\SMATest\\runs\\baseline_s2",
  "max_steps": 12000,
  "warmup": 200,
  "lr_max": 0.0006,
  "lr_min": 6e-05,
  "wd": 0.1,
  "batch_size": 16,
  "grad_accum": 2,
  "block_size": 256,
  "n_layer": 24,
  "n_head": 6,
  "n_embd": 384,
  "vocab_size": 50257,
  "data_seed": 2001343,
  "eval_interval": 1000,
  "eval_batches": 20,
  "log_interval": 20
}