added simple plotting to run

Algo-Boys · Sep 6, 2023 · 9bb4ffb · 9bb4ffb
1 parent e5a003e
commit 9bb4ffb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/swr2_asr/train.py b/swr2_asr/train.py
@@ -11,7 +11,7 @@
 
 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)
         )