Minor fixes

2025-12-09 14:05:33 +00:00 · 2022-02-16 12:41:47 +08:00 · 2022-02-16 12:41:47 +08:00 · d6fefe4e34
commit d6fefe4e34
parent 7cbd6d11ba
1 changed files with 5 additions and 6 deletions
--- a/egs/librispeech/ASR/transducer_stateless_multi_datasets/test_asr_datamodule.py
+++ b/egs/librispeech/ASR/transducer_stateless_multi_datasets/test_asr_datamodule.py
@ -86,12 +86,11 @@ def test_dataset():
            batch_idx += 1

            print("dl idx", idx, "batch_idx", batch_idx)
-            batch = next(dl)
-            cuts = batch["supervisions"]["cut"]
-            for c in cuts:
-                print(c.id)
-
-            if batch_idx > 10:
+            try:
+                _ = next(dl)
+            except StopIteration:
+                print("dl idx", idx)
+                print("Go to the next epoch")
                break