pytorch-wavenet

This is an implementation of the WaveNet architecture, as described in the original paper.

Features

• Automatic creation of a dataset (training and validation/test set) from all sound files (.wav, .aiff, .mp3) in a directory
• Efficient multithreaded data loading
• Logging to TensorBoard (Training loss, validation loss, validation accuracy, parameter and gradient histograms, generated samples)
• Fast generation, as introduced here

Requirements

• python 3
• pytorch 0.3
• numpy
• librosa
• jupyter
• tensorflow for TensorBoard logging

Demo

For an introduction on how to use this model, take a look at the WaveNet demo notebook. You can find audio clips generated by a simple trained model in the generated samples directory