From 102142d1eb59086ee995a137a59540b14c2ffd10 Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 11 Feb 2022 11:05:13 -0700 Subject: [PATCH] f --- codes/trainer/eval/audio_diffusion_fid.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/codes/trainer/eval/audio_diffusion_fid.py b/codes/trainer/eval/audio_diffusion_fid.py index a6f12fb4..9de11787 100644 --- a/codes/trainer/eval/audio_diffusion_fid.py +++ b/codes/trainer/eval/audio_diffusion_fid.py @@ -15,7 +15,7 @@ import trainer.eval.evaluator as evaluator from data.audio.paired_voice_audio_dataset import load_tsv_aligned_codes from data.audio.unsupervised_audio_dataset import load_audio from scripts.audio.gen.speech_synthesis_utils import load_discrete_vocoder_diffuser -from utils.util import opt_get +from utils.util import ceil_multiple, opt_get class AudioDiffusionFid(evaluator.Evaluator): @@ -109,11 +109,11 @@ class AudioDiffusionFid(evaluator.Evaluator): if __name__ == '__main__': - from utils.util import load_model_from_config, ceil_multiple, opt_get + from utils.util import load_model_from_config diffusion = load_model_from_config('X:\\dlas\\experiments\\train_diffusion_tts5_medium.yml', 'generator', also_load_savepoint=False, load_path='X:\\dlas\\experiments\\train_diffusion_tts5_medium\\models\\73000_generator_ema.pth').cuda() opt_eval = {'eval_tsv': 'Y:\\libritts\\test-clean\\transcribed-brief-w2v.tsv', 'diffusion_steps': 50} env = {'rank': 0, 'base_path': 'D:\\tmp\\test_eval', 'step': 500, 'device': 'cuda'} - eval = StyleTransferEvaluator(diffusion, opt_eval, env) + eval = AudioDiffusionFid(diffusion, opt_eval, env) eval.perform_eval() \ No newline at end of file