diff options
| author | Pherkel | 2023-09-01 22:40:29 +0200 |
|---|---|---|
| committer | Pherkel | 2023-09-01 22:40:29 +0200 |
| commit | f3d2ea9a16944434a08e662c5ecfd6ba50e5ea89 (patch) | |
| tree | 01edfa47a0bc84351debd4059d2060bc2d35a687 /swr2_asr/utils.py | |
| parent | 335b8a32f8bba5d37c00af6b4ecd1b9fc520f964 (diff) | |
many todos
Diffstat (limited to 'swr2_asr/utils.py')
| -rw-r--r-- | swr2_asr/utils.py | 4 |
1 files changed, 2 insertions, 2 deletions
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" |