aboutsummaryrefslogtreecommitdiff
path: root/swr2_asr/train.py
diff options
context:
space:
mode:
Diffstat (limited to 'swr2_asr/train.py')
-rw-r--r--swr2_asr/train.py4
1 files changed, 3 insertions, 1 deletions
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])