icewwn / wavernn-1

WaveRNN-based waveform generator & demo of TensorFlow CuDNN-GRU usage.

wavernn-demo

Since this work was part of a private Insight fellowship project, the model structure and training code could not be open-sourced.

This repository contains a demo of a WaveRNN-based waveform generator trained on 13000 spoken sentences; inference can be run on CPU or GPU using the frozen graph.

run_wavernn.py takes an input WAV file, applies an FFT to produce an 80-band mel spectrogram, then uses the spectrogram to generate 16 kHz audio with a frozen WaveRNN model.

cudnn_gru.ipynb demonstrates usage of the (poorly documented) CuDNN GRU cell in TensorFlow, and some of the tricks and workarounds needed to get the network up and running. See this link for an HTML-rendered version.

Setup

Create a virtualenv, activate then run

pip install -r requirements.txt

Run

Choose an input WAV file and run, e.g.

python run_wavernn.py samples/LJ016-0277.wav

CuDNN GRU in TensorFlow - Code and Demonstration

For this model, Nvidia's highly optimized CuDNN GRU trained 7x faster than a slow TensorFlow while-loop with explicit TF operations. Unfortunately, I couldn't get the CudnnGRU module to work out-of-the-box so I resorted to some workarounds to make the cell usable. The Jupyter notebook contains code snippets from the original model, along with an example of how to replicate the CuDNN GRU cell in native TensorFlow.

Requires tensorflow-gpu.