From c0f2abd2a74bd3814d809a5eaa1d3ccfd2e94f47 Mon Sep 17 00:00:00 2001
From: JinZr <60612200+JinZr@users.noreply.github.com>
Date: Fri, 15 Sep 2023 10:58:30 +0800
Subject: [PATCH] minor fix

---
 egs/swbd/ASR/conformer_ctc/sclite_scoring.py | 16 +++++++++++++---
 egs/swbd/ASR/zipformer/sclite_scoring.py     |  6 ++++++
 icefall/checkpoint.py                        |  2 ++
 3 files changed, 21 insertions(+), 3 deletions(-)

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,