From 2f6037861f32ed2721fe66901dafc67aa3cad5b9 Mon Sep 17 00:00:00 2001
From: Marvin Borner
Date: Fri, 18 Aug 2023 14:07:10 +0200
Subject: Fixed @Jojobarthold2 hallo!!!

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

(limited to 'swr2_asr/train.py')

diff --git a/swr2_asr/train.py b/swr2_asr/train.py
index f61776a..df9ffa2 100644
--- a/swr2_asr/train.py
+++ b/swr2_asr/train.py
@@ -1,11 +1,12 @@
 """Training script for the ASR model."""
 from AudioLoader.speech.mls import MultilingualLibriSpeech
+import os
 
 
 def main() -> None:
     """Main function."""
     dataset = MultilingualLibriSpeech(
-        "data", "mls_polish_opus", split="train", download=True
+        "data", "mls_polish_opus", split="train", download=(not os.path.isdir("data"))
     )
 
     print(dataset[1])
-- 
cgit v1.2.3