From 6adf8cd1334129c53e672ba584081235b8767b35 Mon Sep 17 00:00:00 2001
From: Marvin Borner
Date: Fri, 18 Aug 2023 13:32:27 +0200
Subject: Fixed training (use new fork)

---
 swr2_asr/train.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'swr2_asr/train.py')

diff --git a/swr2_asr/train.py b/swr2_asr/train.py
index 34b61ad..abe0e08 100644
--- a/swr2_asr/train.py
+++ b/swr2_asr/train.py
@@ -7,7 +7,9 @@ from AudioLoader.speech.mls import MultilingualLibriSpeech
 
 
 def main():
-    dataset = MultilingualLibriSpeech("./data", "mls_polish_opus", "dev", download=True)
+    dataset = MultilingualLibriSpeech(
+        "data", "mls_polish_opus", split="train", download=True
+    )
 
     print(dataset[1])
 
-- 
cgit v1.2.3