swr2_asr/train.py


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

# import dataset
# define model
# define loss
# define optimizer
# train
from AudioLoader.speech.mls import MultilingualLibriSpeech


def main():
    dataset = MultilingualLibriSpeech(
        "data", "mls_polish_opus", split="train", download=True
    )

    print(dataset[1])


if __name__ == "__main__":
    main()