mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add assertions to mel generator script

Browse Source
This commit is contained in:
James Betker 2022-07-19 11:23:54 -06:00
parent 1b6fe88bcb
commit 4597447178
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
codes/scripts/audio/prep_music/generate_long_mels.py
View File

@ -57,7 +57,10 @@ def process_folder(folder, base_path, output_path, progress_file, max_duration,
total_progress += 30 total_progress += 30
if total_progress > 30: if total_progress > 30:
combined = torch.cat(to_combine, dim=-1).cuda() 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) np.savez(os.path.join(outdir, f'{output_i}'), mel)
output_i += 1 output_i += 1
report_progress(progress_file, folder) report_progress(progress_file, folder)