Skip to content

Example code for GriffinLim Audio transform #1671

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 8 commits into from
Aug 4, 2021
Merged

Example code for GriffinLim Audio transform #1671

Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
3efcbd3
rebased
harishsdev Jun 11, 2021
260ec96
GriffinLim transform example
harishsdev Aug 4, 2021
4d532d3
Merge branch 'main' into GriffinLim
harishsdev Aug 4, 2021
42e9dfe
added modified example code
harishsdev Aug 4, 2021
729ffbe
Merge branch 'GriffinLim' of https://github.com/harishsdev/audio into…
harishsdev Aug 4, 2021
6194f20
modified GriffinLim Example
harishsdev Aug 4, 2021
edb83ed
GriffinLim code modified
harishsdev Aug 4, 2021
bc4742d
modified for tab issues
harishsdev Aug 4, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions torchaudio/transforms.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -155,6 +155,12 @@ class GriffinLim(torch.nn.Module):
Values near 1 can lead to faster convergence, but above 1 may not converge. (Default: ``0.99``)
length (int, optional): Array length of the expected output. (Default: ``None``)
rand_init (bool, optional): Initializes phase randomly if True and to zero otherwise. (Default: ``True``)

Example
>>> batch, freq, time = 2, 257, 100
>>> spectrogram = torch.randn(batch, freq, time)
>>> transform = transforms.GriffinLim(n_fft=512)
>>> waveform = transform(spectrogram)
"""
__constants__ = ['n_fft', 'n_iter', 'win_length', 'hop_length', 'power',
'length', 'momentum', 'rand_init']
Expand Down