diarray commited on
Commit
b703a14
·
verified ·
1 Parent(s): 1e2dcb6
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -60,7 +60,7 @@ img {
60
  The model was fine-tuned using **NVIDIA NeMo** and supports **both TDT (Token-and-Duration Transducer) and CTC (Connectionist Temporal Classification) decoding**.
61
 
62
  ## **🚨 Important Note**
63
- **Update (February 17th):** We observed a significantly lower WER (\~36%) for the TDT branch when using an external WER calculation method that relies solely on the predicted and reference transcriptions. However, the WER values reported in this model card are derived from the standard NeMo workflow using PyTorch Lightning's trainer, where the TDT branch yielded higher WER scores (\~66%). Differences may arise due to variations in post-processing, alignment handling, or evaluation methodologies.
64
 
65
  This model, along with its associated resources, is part of an **ongoing research effort**, improvements and refinements are expected in future versions. Users should be aware that:
66
 
@@ -91,7 +91,7 @@ asr_model = nemo_asr.models.EncDecHybridRNNTCTCBPEModel.from_pretrained(model_na
91
  asr_model.transcribe(['sample_audio.wav'])
92
  ```
93
 
94
- Note that the decoding strategy for the TDT decoder use CUDA Graphs by default but not all GPUs and versions of cuda support that parameter. If you run into a `RuntimeError: CUDA error: invalid argument` you should set that argument to false in the decoding strategy before call asr_model.transcribe()
95
  ```python
96
  decoding_cfg = asr_model.cfg.decoding
97
  # Disable CUDA Graphs
 
60
  The model was fine-tuned using **NVIDIA NeMo** and supports **both TDT (Token-and-Duration Transducer) and CTC (Connectionist Temporal Classification) decoding**.
61
 
62
  ## **🚨 Important Note**
63
+ **Update (February 17th):** We observed a significantly lower WER **(\~36%)** for the TDT branch when using an external WER calculation method that relies solely on the predicted and reference transcriptions. However, the WER values reported in this model card are derived from the standard NeMo workflow using PyTorch Lightning's trainer, where the TDT branch yielded higher WER scores **(\~66%)**. Differences may arise due to variations in post-processing, alignment handling, or evaluation methodologies.
64
 
65
  This model, along with its associated resources, is part of an **ongoing research effort**, improvements and refinements are expected in future versions. Users should be aware that:
66
 
 
91
  asr_model.transcribe(['sample_audio.wav'])
92
  ```
93
 
94
+ Note that the decoding strategy for the TDT decoder use CUDA Graphs by default but not all GPUs and versions of cuda support that parameter. If you run into a `RuntimeError: CUDA error: invalid argument` you should set that argument to false in the decoding strategy before calling asr_model.transcribe()
95
  ```python
96
  decoding_cfg = asr_model.cfg.decoding
97
  # Disable CUDA Graphs