From f3d2ea9a16944434a08e662c5ecfd6ba50e5ea89 Mon Sep 17 00:00:00 2001 From: Pherkel Date: Fri, 1 Sep 2023 22:40:29 +0200 Subject: many todos --- swr2_asr/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'swr2_asr/utils.py') diff --git a/swr2_asr/utils.py b/swr2_asr/utils.py index 4c751d5..3b9b3ca 100644 --- a/swr2_asr/utils.py +++ b/swr2_asr/utils.py @@ -236,6 +236,7 @@ class MLSDataset(Dataset): book_id=self.dataset_lookup[idx]["bookid"], chapter_id=self.dataset_lookup[idx]["chapterid"], ) + def collate_fn(samples: list[Sample]) -> dict: """Collate function for the dataloader @@ -266,8 +267,7 @@ def collate_fn(samples: list[Sample]) -> dict: "utterance": labels, "utterance_length": label_lengths, } - - + if __name__ == "__main__": dataset_path = "/Volumes/pherkel/SWR2-ASR" -- cgit v1.2.3