diff options
| author | Marvin Borner | 2023-08-18 13:32:27 +0200 |
|---|---|---|
| committer | Marvin Borner | 2023-08-18 13:32:40 +0200 |
| commit | 6adf8cd1334129c53e672ba584081235b8767b35 (patch) | |
| tree | 462f7dc327ea1578c3cfa2f18fb13639467fee2a /swr2_asr/train.py | |
| parent | e28c8327baad8af6893c816d127593d1a9ab3609 (diff) | |
Fixed training (use new fork)
Diffstat (limited to 'swr2_asr/train.py')
| -rw-r--r-- | swr2_asr/train.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/swr2_asr/train.py b/swr2_asr/train.py index 34b61ad..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_polish_opus", "dev", download=True) + dataset = MultilingualLibriSpeech( + "data", "mls_polish_opus", split="train", download=True + ) print(dataset[1]) |