Reformat by black non-streaming zipformer recipe for ksponspeech (#1665)

k2-fsa · Jun 24, 2024 · 031f892 · 031f892
1 parent 6f102d3
commit 031f892
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 8 deletions.
diff --git a/egs/ksponspeech/ASR/zipformer/ctc_decode.py b/egs/ksponspeech/ASR/zipformer/ctc_decode.py
@@ -571,7 +571,9 @@ def save_results(
         # ref/hyp pairs.
         errs_filename = params.res_dir / f"errs-{test_set_name}-{params.suffix}.txt"
         with open(errs_filename, "w") as f:
-            cer = write_error_stats(f, f"{test_set_name}-{key}", results, compute_CER=True)
+            cer = write_error_stats(
+                f, f"{test_set_name}-{key}", results, compute_CER=True
+            )
             test_set_cers[key] = cer
 
         logging.info("Wrote detailed error stats to {}".format(errs_filename))
@@ -807,15 +809,15 @@ def main():
 
     # we need cut ids to display recognition results.
     args.return_cuts = True
-    
+
     ksponspeech = KsponSpeechAsrDataModule(args)
 
     eval_clean_cuts = ksponspeech.eval_clean_cuts()
     eval_other_cuts = ksponspeech.eval_other_cuts()
 
     eval_clean_dl = ksponspeech.test_dataloaders(eval_clean_cuts)
     eval_other_dl = ksponspeech.test_dataloaders(eval_other_cuts)
-    
+
     test_sets = ["eval_clean", "eval_other"]
     test_dl = [eval_clean_dl, eval_other_dl]
 

diff --git a/egs/ksponspeech/ASR/zipformer/decode.py b/egs/ksponspeech/ASR/zipformer/decode.py
@@ -727,7 +727,11 @@ def save_results(
         )
         with open(errs_filename, "w") as f:
             cer = write_error_stats(
-                f, f"{test_set_name}-{key}", results, enable_log=True, compute_CER=True,
+                f,
+                f"{test_set_name}-{key}",
+                results,
+                enable_log=True,
+                compute_CER=True,
             )
             test_set_cers[key] = cer
 

diff --git a/egs/ksponspeech/ASR/zipformer/streaming_decode.py b/egs/ksponspeech/ASR/zipformer/streaming_decode.py
@@ -659,7 +659,11 @@ def save_results(
         )
         with open(errs_filename, "w") as f:
             cer = write_error_stats(
-                f, f"{test_set_name}-{key}", results, enable_log=True, compute_CER=True,
+                f,
+                f"{test_set_name}-{key}",
+                results,
+                enable_log=True,
+                compute_CER=True,
             )
             test_set_cers[key] = cer
 

diff --git a/egs/ksponspeech/ASR/zipformer/train.py b/egs/ksponspeech/ASR/zipformer/train.py
@@ -961,9 +961,7 @@ def save_bad_model(suffix: str = ""):
             scaler.update()
             optimizer.zero_grad()
         except Exception as e:
-            logging.info(
-                f"Caught exception: {e}."
-            )
+            logging.info(f"Caught exception: {e}.")
             save_bad_model()
             display_and_save_batch(batch, params=params, sp=sp)
             raise