diff --git a/codes/scripts/audio/mel_bin_norm_compute.py b/codes/scripts/audio/mel_bin_norm_compute.py index e2f942b7..663da801 100644 --- a/codes/scripts/audio/mel_bin_norm_compute.py +++ b/codes/scripts/audio/mel_bin_norm_compute.py @@ -31,4 +31,4 @@ if __name__ == '__main__': if len(mels) > args.num_batches: break mel_norms = torch.stack(mels).mean(0) - torch.save('mel_norms.pth', mel_norms) \ No newline at end of file + torch.save(mel_norms, 'mel_norms.pth') \ No newline at end of file