Minor fixes

This commit is contained in:
Fangjun Kuang 2022-02-16 12:41:47 +08:00
parent 7cbd6d11ba
commit d6fefe4e34

View File

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