From 459744717867feba46a750743aac25cac0835561 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 19 Jul 2022 11:23:54 -0600 Subject: [PATCH] add assertions to mel generator script --- codes/scripts/audio/prep_music/generate_long_mels.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/codes/scripts/audio/prep_music/generate_long_mels.py b/codes/scripts/audio/prep_music/generate_long_mels.py index 1f14fe4a..19dd0bea 100644 --- a/codes/scripts/audio/prep_music/generate_long_mels.py +++ b/codes/scripts/audio/prep_music/generate_long_mels.py @@ -57,7 +57,10 @@ def process_folder(folder, base_path, output_path, progress_file, max_duration, total_progress += 30 if total_progress > 30: combined = torch.cat(to_combine, dim=-1).cuda() - mel = produce_mel(combined).cpu().numpy() + mel = produce_mel(combined) + assert mel.max() < 1.00001, mel.max() + assert mel.min() > -1.00001, mel.min() + mel = mel.cpu().numpy() np.savez(os.path.join(outdir, f'{output_i}'), mel) output_i += 1 report_progress(progress_file, folder)