From 9bb4ffbd0ef84bfea428112495cc63266e3b0955 Mon Sep 17 00:00:00 2001
From: JoJoBarthold2
Date: Wed, 6 Sep 2023 13:09:48 +0200
Subject: added simple plotting to run

---
 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 40626e7..9f12bcb 100644
--- a/swr2_asr/train.py
+++ b/swr2_asr/train.py
@@ -11,7 +11,7 @@ from tqdm import tqdm
 
 from swr2_asr.model_deep_speech import SpeechRecognitionModel
 from swr2_asr.tokenizer import CharTokenizer, train_char_tokenizer
-from swr2_asr.utils import MLSDataset, Split, collate_fn
+from swr2_asr.utils import MLSDataset, Split, collate_fn,plot
 
 from .loss_scores import cer, wer
 
@@ -268,6 +268,7 @@ def run(
                 "avg_wer": avg_wer,
             },
             path + str(epoch),
+    plot(epochs,path)
         )
 
 
-- 
cgit v1.2.3