{
  "batch_size": 128,
  "layers": 12,
  "heads": 8,
  "dim": 512,
  "lr_factor": 0.1,
  "enable_flash": true,
  "enable_grad_scaler": false,
  "quantize_divisions": 24,
  "pitch_min": 24,
  "pitch_max": 96,
  "length_min": 1,
  "length_max": 144,
  "polyphony_max": 5,
  "embed_length_max": 768,
  "embed_polyphony_max": 8,
  "aug_transpose_range": 6,
  "aug_rebar_max_bars": 2,
  "aug_rebar_prob": 0.1,
  "seq_len": 400,
  "seq_len_min": 96,
  "label_smoothing": 0.0,
  "weight_decay": 0,
  "accum_iter": 1,
  "epochs": 1,
  "max_examples_per_rank": 128000,
  "enable_eval": false,
  "enable_checkpoint": false
}