shuokay / mxnet-wavenet

MXNet implementation of WaveNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MXNet Implementation of WaveNet

I am trying to reproduce the WaveNet result using MXNet. Here is the training code of generating without any condition, but the training process can't convergence since the mae is always around 126. Hoping someone can raise your advices. The model convergence in fact. The reason of "mae is always around 126" is the mxnet office mae evaluation metric does not match this net config. I have defined a new EvalMetric
Training log

How to Run

  1. Install mxnet and fix the dilate bug according to apache/mxnet#3479
  2. Download VCTK cprpus and extract to the root folder
  3. Start training by python train.py

Implementation Note

  • Padding zero on right side by Concat operator since the convolution op of mxnet can't pad only on one side.

About

MXNet implementation of WaveNet


Languages

Language:Python 100.0%