diff --git a/egs/swbd/ASR/conformer_ctc/sclite_scoring.py b/egs/swbd/ASR/conformer_ctc/sclite_scoring.py index fe0fbbeee..0383c4d71 100755 --- a/egs/swbd/ASR/conformer_ctc/sclite_scoring.py +++ b/egs/swbd/ASR/conformer_ctc/sclite_scoring.py @@ -81,13 +81,23 @@ def asr_text_post_processing(text: str) -> str: remaining_words.append("H") remaining_words.append("D") continue + elif word == "UCLA": + remaining_words.append("U") + remaining_words.append("C") + remaining_words.append("L") + remaining_words.append("A") + continue elif word == "ONTO": remaining_words.append("ON") remaining_words.append("TO") continue - elif word == "DAY" and text_split[idx + 1] == "CARE": - remaining_words.append("DAYCARE") - word_to_skip = 1 + elif word == "DAY": + try: + if text_split[idx + 1] == "CARE": + remaining_words.append("DAYCARE") + word_to_skip = 1 + except: + remaining_words.append(word) continue remaining_words.append(word) diff --git a/egs/swbd/ASR/zipformer/sclite_scoring.py b/egs/swbd/ASR/zipformer/sclite_scoring.py index 239204221..0383c4d71 100755 --- a/egs/swbd/ASR/zipformer/sclite_scoring.py +++ b/egs/swbd/ASR/zipformer/sclite_scoring.py @@ -81,6 +81,12 @@ def asr_text_post_processing(text: str) -> str: remaining_words.append("H") remaining_words.append("D") continue + elif word == "UCLA": + remaining_words.append("U") + remaining_words.append("C") + remaining_words.append("L") + remaining_words.append("A") + continue elif word == "ONTO": remaining_words.append("ON") remaining_words.append("TO") diff --git a/icefall/checkpoint.py b/icefall/checkpoint.py index c83c56a53..f7712628b 100644 --- a/icefall/checkpoint.py +++ b/icefall/checkpoint.py @@ -76,7 +76,9 @@ def save_checkpoint( if isinstance(model, DDP): model = model.module + # import pdb + # pdb.set_trace() checkpoint = { "model": model.state_dict(), "optimizer": optimizer.state_dict() if optimizer is not None else None,