wenetspeech

This commit is contained in:
游雁 2024-05-22 16:20:24 +08:00
parent b37e7b15e0
commit 0b764d3350

View File

@ -75,6 +75,7 @@ def main(**kwargs):
if batch_idx % log_step == 0:
print(f"prcessed: {batch_idx}/{iter_stop}")
if batch_idx >= iter_stop and iter_stop > 0.0:
print(f"prcessed: {iter_stop}/{iter_stop}")
break
fbank = batch["speech"].numpy()[0, :, :]