swr2_asr/train.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

"""Training script for the ASR model."""
import os
from AudioLoader.speech.mls import MultilingualLibriSpeech


def main() -> None:
    """Main function."""
    dataset = MultilingualLibriSpeech(
        "data", "mls_polish_opus", split="train", download=(not os.path.isdir("data"))
    )

    print(dataset[1])


if __name__ == "__main__":
    main()