swr2_asr/train.py


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

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


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

    print(dataset[1])


if __name__ == "__main__":
    main()