Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ASR]:fixed augmentor arguments for transcribe functionality of Hybrid CTC-RNNT model #6290

Merged
merged 1 commit into from
Mar 24, 2023

Conversation

KunalDhawan
Copy link
Collaborator

What does this PR do ?

Fixes issues in nemo/collections/asr/models/hybrid_rnnt_ctc_models.py regarding Augmentor parameters to enable transcribe functionality for hybrid RNNT-CTC models

Collection: [ASR]

Changelog

  • in the function transcribe() of class EncDecHybridRNNTCTCModel, added
    • augmentor argument in function call (line 103)
    • augmentor argument in the config for calling _setup_transcribe_dataloader() (line 179)

Usage

  • You can potentially add a usage example below
python  NeMo/examples/asr/transcribe_speech.py \
    model_path="path to hybrid model" \
    pretrained_name="name of hybrid model from NGC directory" \
    audio_dir="path to folder of audio files" \
    dataset_manifest="path to manifest" \
    output_filename="output filename" \
    batch_size=32 \
    compute_timestamps=False \
    compute_langs=False \
    cuda=0 \
    amp=True \
    append_pred=False \
    pred_name_postfix="another model name for output filename, if required"

Before your PR is "Ready for review"

Pre checks:

  • Make sure you read and followed Contributor guidelines
  • Did you write any new necessary tests?
  • Did you add or update any necessary documentation?
  • Does the PR affect components that are optional to install? (Ex: Numba, Pynini, Apex etc)
    • Reviewer: Does the PR have correct import guards for all optional libraries?

PR Type:

  • New Feature
  • Bugfix
  • Documentation

If you haven't finished some of the above items you can still open "Draft" PR.

Who can review?

Anyone in the community is free to review the PR once the checks have passed.
Contributor guidelines contains specific people who can review PRs to various areas.

Additional Information

  • Related to # (issue)

@KunalDhawan KunalDhawan force-pushed the fix_transcribe_asr_hybrid_model branch from caee365 to d59c169 Compare March 24, 2023 19:45
@titu1994 titu1994 merged commit 0ced355 into r1.17.0 Mar 24, 2023
@titu1994 titu1994 deleted the fix_transcribe_asr_hybrid_model branch March 24, 2023 21:59
github-actions bot pushed a commit that referenced this pull request Mar 24, 2023
titu1994 pushed a commit that referenced this pull request Mar 24, 2023
… CTC-RNNT model (#6290) (#6292)

Signed-off-by: KunalDhawan <[email protected]>
Co-authored-by: Kunal Dhawan <[email protected]>
hsiehjackson pushed a commit to hsiehjackson/NeMo that referenced this pull request Jun 2, 2023
… CTC-RNNT model (NVIDIA#6290) (NVIDIA#6292)

Signed-off-by: KunalDhawan <[email protected]>
Co-authored-by: Kunal Dhawan <[email protected]>
Signed-off-by: hsiehjackson <[email protected]>
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.

None yet

3 participants