This commit is contained in:
qmpzzpmq 2022-08-15 15:12:45 +00:00
parent 6da35058fa
commit 74ee6e6130

View File

@ -105,7 +105,7 @@ def generate_ctc_label_batch(
supervisions = batch["supervisions"]
nnet_output, memory, memory_key_padding_mask = model(feature, supervisions)
return nnet_output
def generate_ctc_label_dataset(
dl: torch.utils.data.DataLoader,