Skip to content

Comments

[Draft] [TTS] Evaluation: Batch the ASR; refactor for parallelization#15417

Draft
rfejgin wants to merge 20 commits intoNVIDIA-NeMo:mainfrom
rfejgin:magpietts_evaluation_parallelization
Draft

[Draft] [TTS] Evaluation: Batch the ASR; refactor for parallelization#15417
rfejgin wants to merge 20 commits intoNVIDIA-NeMo:mainfrom
rfejgin:magpietts_evaluation_parallelization

Conversation

@rfejgin
Copy link
Collaborator

@rfejgin rfejgin commented Feb 20, 2026

No description provided.

For debugging NeMo Skills deployment

Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
When running the project from a directory different from the repo root
(which happens in NeMo Skills), these paths need to be converted to absolute
paths, which is done in this commit.

Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
… in Nemo Skills)

Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
@github-actions github-actions bot added the TTS label Feb 20, 2026
- remove g2p path handling (unrelated to parallelization)
- update a comment

Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
It has been moved outside of evaluate() since in the NeMo Skills use
case we need the full metrics for chunk-wise scoring and aggregation
at the end.

Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
@rfejgin rfejgin changed the title [TTS] Infrastructure for parallel evalution [TTS] Parallel evolution infrastructure Feb 21, 2026
@rfejgin rfejgin changed the title [TTS] Parallel evolution infrastructure [Draft] [TTS] Parallel evolution infrastructure Feb 21, 2026
@rfejgin rfejgin changed the title [Draft] [TTS] Parallel evolution infrastructure [Draft] [TTS] Parallel evaluation infrastructure Feb 21, 2026
- Break evaluation into two steps:
  - evaluate_dir() for directory-level evaluation. Can be run in parallel across multiple directories (e.g. in NeMo Skills)
  - compute_global_metrics() for global metrics aggregation.
- Move model loading to separate function
- Cleanup

Signed-off-by: Fejgin, Roy <rfejgin@nvidia.com>
@rfejgin rfejgin changed the title [Draft] [TTS] Parallel evaluation infrastructure [Draft] [TTS] Evaluation: Batch the ASR; refactor for parallelization Feb 21, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant