diff options
Diffstat (limited to 'swr2_asr/train.py')
| -rw-r--r-- | swr2_asr/train.py | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/swr2_asr/train.py b/swr2_asr/train.py index abe0e08..f61776a 100644 --- a/swr2_asr/train.py +++ b/swr2_asr/train.py @@ -1,12 +1,9 @@ -# import dataset -# define model -# define loss -# define optimizer -# train +"""Training script for the ASR model.""" from AudioLoader.speech.mls import MultilingualLibriSpeech -def main(): +def main() -> None: + """Main function.""" dataset = MultilingualLibriSpeech( "data", "mls_polish_opus", split="train", download=True ) |