config_stage1.json

{
  "cv_version": "fsicoli/twi",
  "cv_lang": "tw",
  "split_train": "train",
  "split_eval": "validation",
  "sample_rate": 16000,
  "max_audio_seconds": 30.0,
  "whisper_ckpt": "femursmith/intermediate-asr-ashanti-twi",
  "batch_size": 2,
  "num_workers": 0,
  "label_max_length": 256,
  "rvq_dim": 1024,
  "rvq_num_quantizers": 12,
  "rvq_codebook_size": 2048,
  "rvq_commitment_weight": 0.25,
  "lr": 0.0003,
  "epochs": 3,
  "warmup_steps": 200,
  "beta_max": 0.7,
  "beta_ramp": 1000,
  "save_every": 1000,
  "eval_every": 400,
  "gen": {
    "num_beams": 6,
    "length_penalty": 1.1,
    "no_repeat_ngram_size": 3
  }
}