Updates for Sortformer training - #16227

Merged

nithinraok merged 17 commits into

NVIDIA-NeMo:mainfrom

ipmedenn:update-sortformer-training

Sep 12, 2026

Commits on Sep 2, 2026

updating sorrtformer training scripts, adding ckpt averaging
ipmedenn
committed
7d3f950
updating nemo dataloader for sortformer
ipmedenn
committed
496edf8

Commits on Sep 3, 2026

adding OversamplingDistributedSampler
ipmedenn
committed
b36fde0
adding yaml configs for 8spk Sortformer
ipmedenn
committed
b72ab5d

Commits on Sep 4, 2026

logits based BCE loss for sortformer
ipmedenn
committed
08667f8

Commits on Sep 5, 2026

auxiliary losses for sortformer
ipmedenn
committed
7c78a29
Fix Sortformer scheduler max_steps on resume
ipmedenn
committed
2f4b6ec
Merge NVIDIA-NeMo/Speech main into update-sortformer-training
ipmedenn
committed
f4b5a52
Add block-causal tail augmentation to streaming Sortformer training
ipmedenn
committed
efa1bc1
Add speaker counting metrics for Sortformer training and validation
ipmedenn
committed
99da565

Commits on Sep 6, 2026

Optimize OOM-safe Sortformer feature extraction
ipmedenn
committed
6fb624e
Add Hugging Face model loading to Sortformer diarization
ipmedenn
committed
8108f6a
Update 8-speaker Sortformer config comments and license header
ipmedenn
committed
bcd71bc

Commits on Sep 7, 2026

Actualizing sortformer_streaming_8spk.yaml
ipmedenn
committed
4b5d1bf
Addressing CodeQL issues
ipmedenn
committed
a2924f4

Commits on Sep 8, 2026

Organize Sortformer auxiliary losses and metrics
ipmedenn
committed
d2ee5e0

Commits on Sep 11, 2026

Refactor auxiliary diarization losses into configurable classes
ipmedenn
committed
f497938