NTT123 / fast_wavegru

Fast C++ WaveGRU

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Introduction

We use DeepMind's sparse matmul library to speed-up WaveGRU inference.

Build

$ go get github.com/bazelbuild/bazelisk
$ sudo apt-get install libsndfile1 -y
$ pip install librosa
$ bazelisk build wavegru -c opt --copt=-march=native

Run

$ ./bazel-bin/wavegru --weight ./weight.pickle --mel ./ft.npy --output audio.wav

About

Fast C++ WaveGRU

License:Apache License 2.0


Languages

Language:C++ 95.9%Language:Starlark 3.6%Language:Python 0.4%Language:C 0.2%