Skip to content

dhgrs/chainer-VQ-VAE

Repository files navigation

chainer-VQ-VAE

A Chainer implementation of VQ-VAE( https://arxiv.org/abs/1711.00937 ).

Results

Trained about 63 hours with one 1080Ti (150000 iterations) on VCTK-Corpus. You can download pretrained model from here.

Losses:

loss1 loss2 loss3

Audios:

demo

Requirements

I trained and generated with

  • python(3.5.2)
  • chainer(4.0.0b3)
  • librosa(0.5.1)

And now you can try it on Google Colaboratory. You don't need install chainer/librosa in your local or buy GPUs. Check this.

Usage

download dataset

You can download VCTK-Corpus(en) from here. And you can download CMU-ARCTIC(en)/voice-statistics-corpus(ja) very easily via my repository.

set parameters

parameters of training

  • batchsize
    • Batch size.
  • lr
    • Learning rate.
  • ema_mu
    • Rate of exponential moving average. If this is greater than 1 doesn't apply.
  • trigger
    • How many times you update the model. You can set this parameter like as (<int>, 'iteration') or (<int>, 'epoch')
  • evaluate_interval
    • The interval that you evaluate validation dataset. You can set this parameter like as trigger.
  • snapshot_interval
    • The interval that you save snapshot. You can set this parameter like as trigger.
  • report_interval
    • The interval that you write log of loss. You can set this parameter like as trigger.

parameters of dataset

  • root
    • The root directory of training dataset.
  • dataset
    • The architecture of the directory of training dataset. Now this parameter supports VCTK, ARCTIC and 'vs'.
  • split_seed
    • A seed for splitting dataset into train and validation.

parameters of preprocessing

  • sr
    • Sampling rate. If it's different from input file, be resampled by librosa.
  • res_type
    • The resampling algorithm used in librosa.
  • top_db
    • The threshold db for triming silence.
  • input_dim
    • The input channels of wave. If it is 1, mu-law is not applied. Else mu-law is applied.
  • quantize
    • The number for quantize.
  • length
    • How many samples used for training.
  • use_logistic
    • Use mixture of logistics or not.

parameters of VQ

  • d
    • The parameter d in the paper.
  • k
    • The parameter k in the paper.

parameters of Decoder(WaveNet)

  • n_loop
    • If you want to make network like dilations [1, 2, 4, 1, 2, 4] set n_loop as 2.
  • n_layer
    • If you want to make network like dilations [1, 2, 4, 1, 2, 4] set n_layer as 3.
  • filter_size
    • The filter size of each dilated convolution.
  • residual_channels
    • The number of input/output channels of residual blocks.
  • dilated_channels
    • The number of output channels of causal dilated convolution layers. This is splited into tanh and sigmoid so the number of hidden units is half of this number.
  • skip_channels
    • The number of channels of skip connections and last projection layer.
  • n_mixture
    • The number of logistic distribution. It is used only use_logistic is True.
  • log_scale_min
    • The number for stability. It is used only use_logistic is True.
  • global_condition_dim
    • The dimension of speaker embeded-vector.
  • local_condition_dim
    • The dimension of local contioning vectors.
  • dropout_zero_rate
    • The rate of 0 in dropout. If 0 doesn't apply dropout.

parameters of losses

  • beta
    • The parameter beta in the paper.

parameters of losses

  • use_ema
    • If True use the value of exponential moving average.
  • apply_dropout
    • If True apply dropout.

training

(without GPU)
python train.py

(with GPU #n)
python train.py -g n

If you want to use multi GPUs, you can add IDs like below.

python train.py -g 0 1 2

You can resume snapshot and restart training like below.

python train.py -r snapshot_iter_100000

Other arguments -f and -p are parameters for multiprocess in preprocessing. -f means the number of prefetch and -p means the number of processes.

generating

python generate.py -i <input file> -o <output file> -m <trained model> -s <speaker>

If you don't set -o, default file name result.wav is used. If you don't set -s, the speaker is same as input file that got from filepath.

TODO

  • upload generated sample
  • using GPU fot generating
  • descritized mixture of logistics
  • Parallel WaveNet

Releases

No releases published

Packages

No packages published