huseinzolkepliscicom commited on
Commit
a65255f
·
verified ·
1 Parent(s): 301dc36

README: document stage-1 (noisy) revision + stage-2 latest

Browse files
Files changed (1) hide show
  1. README.md +18 -0
README.md CHANGED
@@ -37,6 +37,24 @@ Load instructions (canonical):
37
  | `pytorch_model.bin` | decoder weights for **inference** (load with `NeuCodec._from_pretrained(..., decoder_depth=20)`) |
38
  | `last.ckpt` | full PyTorch-Lightning checkpoint **with optimizer states + LR schedulers** for **resuming training** |
39
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
40
  ## Training data
41
 
42
  Diverse, predominantly high-sample-rate human speech (no synthetic TTS). The **scale-up group is
 
37
  | `pytorch_model.bin` | decoder weights for **inference** (load with `NeuCodec._from_pretrained(..., decoder_depth=20)`) |
38
  | `last.ckpt` | full PyTorch-Lightning checkpoint **with optimizer states + LR schedulers** for **resuming training** |
39
 
40
+ ## Training stages / which revision to pull
41
+
42
+ This model is trained in two stages. `main` always points at the **latest** stage.
43
+
44
+ | stage | data | how to load |
45
+ |---|---|---|
46
+ | **Stage 1 — noisy mix** (steps 0→1.58M, `epoch=9-step=1580000`) | 8 base corpora + `scale44k` (~6,500 h, mostly noisy crowdsourced/podcast) | pin revision **`d2c612f4319c1c1e0c65a9431a7d6db1b7955ce0`** |
47
+ | **Stage 2 — clean finetune** (from 1.58M, clean ≥44.1 kHz only; in progress) | TTS-Clean44k + Clean-Podcast + clean-teacher pool + EARS/Expresso | `main` (default) |
48
+
49
+ ```python
50
+ # Stage 1 (first-stage noisy training) — pin the exact revision:
51
+ model = NeuCodec._from_pretrained(
52
+ model_id="Scicom-intl/neucodec-44k-d20",
53
+ decoder_depth=20,
54
+ revision="d2c612f4319c1c1e0c65a9431a7d6db1b7955ce0",
55
+ )
56
+ ```
57
+
58
  ## Training data
59
 
60
  Diverse, predominantly high-sample-rate human speech (no synthetic TTS). The **scale-up group is