Upload 24k/config.yaml with huggingface_hub

24k/config.yaml

@@ -0,0 +1,46 @@
+# 24 kHz production config (STFT mask path). Matches the trained 24k checkpoint.
+seed: 0
+device: auto
+
+stft:
+  sample_rate: 24000
+  n_fft: 2048
+  hop_length: 300
+  win_length: 1200
+  log_eps: 1.0e-5
+
+model:
+  base_channels: 32
+  depth: 4
+  gru_hidden: 256
+  gru_reduce: 32
+
+norm:
+  mean: -3.314941
+  std: 2.795372
+
+aug:
+  snr_db: [0.0, 20.0]
+  reverb_prob: 0.5
+  noise_prob: 1.0
+  remove_reverb: false
+
+data:
+  clean_dirs: [data/clean_24k]   # e.g. LJSpeech + JVS resampled to 24k
+  noise_dir: data/noise
+  rir_dir: data/rir
+  segment_frames: 128
+  val_fraction: 0.05
+  num_workers: 16
+  dummy: false
+  dummy_size: 64
+
+train:
+  batch_size: 16
+  lr: 2.0e-4
+  max_steps: 200000
+  log_every: 100
+  val_every: 2000
+  ckpt_every: 2000
+  ckpt_dir: outputs/ckpt_24k
+  amp: true