This commit is contained in:
James Betker 2021-12-11 08:18:00 -07:00
parent 5a664aa56e
commit 959979086d

View File

@ -31,4 +31,4 @@ if __name__ == '__main__':
if len(mels) > args.num_batches: if len(mels) > args.num_batches:
break break
mel_norms = torch.stack(mels).mean(0) mel_norms = torch.stack(mels).mean(0)
torch.save('mel_norms.pth', mel_norms) torch.save(mel_norms, 'mel_norms.pth')