diff options
Diffstat (limited to 'swr2_asr/train.py')
| -rw-r--r-- | swr2_asr/train.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/swr2_asr/train.py b/swr2_asr/train.py index f61776a..df9ffa2 100644 --- a/swr2_asr/train.py +++ b/swr2_asr/train.py @@ -1,11 +1,12 @@ """Training script for the ASR model.""" from AudioLoader.speech.mls import MultilingualLibriSpeech +import os def main() -> None: """Main function.""" dataset = MultilingualLibriSpeech( - "data", "mls_polish_opus", split="train", download=True + "data", "mls_polish_opus", split="train", download=(not os.path.isdir("data")) ) print(dataset[1]) |