diff options
| -rw-r--r-- | .gitignore | 4 | ||||
| -rw-r--r-- | swr2_asr/train.py | 4 |
2 files changed, 6 insertions, 2 deletions
@@ -170,4 +170,6 @@ cython_debug/ # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore # and can be added to the global gitignore or merged into this file. For a more nuclear # option (not recommended) you can uncomment the following to ignore the entire idea folder. -#.idea/
\ No newline at end of file +#.idea/ + +data/ diff --git a/swr2_asr/train.py b/swr2_asr/train.py index cfdef6b..abe0e08 100644 --- a/swr2_asr/train.py +++ b/swr2_asr/train.py @@ -7,7 +7,9 @@ from AudioLoader.speech.mls import MultilingualLibriSpeech def main(): - dataset = MultilingualLibriSpeech("./data", "mls_german_opus", "dev", download=True) + dataset = MultilingualLibriSpeech( + "data", "mls_polish_opus", split="train", download=True + ) print(dataset[1]) |