aboutsummaryrefslogtreecommitdiff
path: root/swr2_asr/train.py
diff options
context:
space:
mode:
authorMarvin2023-08-18 12:59:17 +0200
committerGitHub2023-08-18 12:59:17 +0200
commitff574af3cfb4ebfc2c1faf8cdc4bf87560dd8373 (patch)
treecb4f912bc3a9a2c5e8ba2b8567fe2c8c785cf661 /swr2_asr/train.py
parent58f07302989f945b109c0c69d9e96444514ef1a6 (diff)
parent2bcaf9c687dfc3a19d0409361a3dc9b535150e8a (diff)
Merge pull request #7 from Algo-Boys/feature-dataloading
Feature dataloading
Diffstat (limited to 'swr2_asr/train.py')
-rw-r--r--swr2_asr/train.py16
1 files changed, 16 insertions, 0 deletions
diff --git a/swr2_asr/train.py b/swr2_asr/train.py
index e69de29..cfdef6b 100644
--- a/swr2_asr/train.py
+++ b/swr2_asr/train.py
@@ -0,0 +1,16 @@
+# import dataset
+# define model
+# define loss
+# define optimizer
+# train
+from AudioLoader.speech.mls import MultilingualLibriSpeech
+
+
+def main():
+ dataset = MultilingualLibriSpeech("./data", "mls_german_opus", "dev", download=True)
+
+ print(dataset[1])
+
+
+if __name__ == "__main__":
+ main()