RecorrNet-v2 — AISHELL-1 Mix ver1 epoch 7
Single-channel RecorrNet-v2 (direct 9-tap filter head) fine-tuned on AISHELL-1 Mix ver1 (variable 1–5 speakers, mix_clean + mix_both).
- Architecture: RecorrNet-v2 (
1ch_aishell1mix_var_2_5spk_recorr_v2) - Init: English WSJ0 variable 2–5spk SR-CorrNet
- Checkpoint:
epoch.0007.pth(end of epoch 7; Aishell-only, not AliMeeting continue) - Code: vu-duy-tung/byd-speech-separation (
recorrnetbranch)
This is the RecorrNet epoch-7 ver1 checkpoint used in the n100 SR vs Recorr vs later AliMeeting-continue comparisons.
Load
from sr_corrnet import SSInference
model = SSInference.from_pretrained(
"playwithmino/recorrnet-v2-1ch-aishell1mix-ver1-epoch7",
device="cuda:0",
)
result = model.process_file("mix.wav", output_dir="output/")
Files in this repo: model.pt (weights only) and config.yaml.