/wavegan-torch

PyTorch reimplementation of WaveGAN

Primary LanguageJupyter Notebook

WaveGAN

A PyTorch reimplementation of WaveGAN (Donahue, et al. 2018).

Setup

This code requires Python 3 and ffmpeg (which can be installed with conda), along with the following packages (which can be installed with conda or pip):

  • pytorch
  • numpy
  • librosa
  • pescador (for sampling)

Running

You can train the WaveGAN with audio from a given directory by using train_wavegan.py. To run with default parameters, run:

python train_wavegan.py <audio_dir> <output_dir>

To see the full list of arguments, run python train_wavegan.py -h. Note that the data split for validation and testing is done on a filewise basis. For SLURM users, an example sbatch script is provided in jobs/train-wavegan.sbatch.