run_scorer.py

"""
Example use of the scorer utility to inspect data.

(pre-)trained models and already cache files with extracted features are required.
"""
import os
from Utility.Scorer import AlignmentScorer
from Utility.Scorer import TTSScorer
from Utility.storage_config import MODELS_DIR, PREPROCESSING_DIR
import torch

exec_device = "cuda" if torch.cuda.is_available() else "cpu"

alignment_scorer = AlignmentScorer(path_to_aligner_model=os.path.join(MODELS_DIR, "Aligner", "aligner.pt"), device=exec_device)
alignment_scorer.score(path_to_aligner_dataset=os.path.join(PREPROCESSING_DIR, "IntegrationTest", "aligner_train_cache.pt"))
alignment_scorer.show_samples_with_highest_loss(20)

tts_scorer = TTSScorer(path_to_fastspeech_model=os.path.join(MODELS_DIR, "FastSpeech2_IntegrationTest","best.pt"), device=exec_device)
tts_scorer.score(path_to_fastspeech_dataset=os.path.join(PREPROCESSING_DIR, "IntegrationTest/"), lang_id="en")
tts_scorer.show_samples_with_highest_loss(20)
tts_scorer.remove_samples_with_highest_loss(20)