diff options
Diffstat (limited to 'swr2_asr')
| -rw-r--r-- | swr2_asr/train.py | 3 |
1 files changed, 2 insertions, 1 deletions
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) ) |