Skip to content
Wavenet Autoencoder for Unsupervised speech representation learning (after Chorowski, Jan 2019)
Branch: master
Clone or download
Latest commit f310210 Apr 17, 2019
Type Name Latest commit message Commit time
Failed to load latest commit information.
ae_wavenet.ipynb almost working rfield Apr 2, 2019 progress fixing sgvb Apr 18, 2019 maybe fixed checkpointing Apr 13, 2019 actual training? Apr 14, 2019
notes.txt adding sgvb and vae Apr 17, 2019
padding_notes.txt jitter, cuda-agnostic Apr 7, 2019 maybe fixed checkpointing Apr 13, 2019 extra metrics, cleaner checkpointing Apr 11, 2019
rfield_notes.txt rfield extra porcelain Apr 5, 2019
todo.txt actual training? Apr 14, 2019
upsampling_notes.txt actual training? Apr 14, 2019

PyTorch implementation of Jan Chorowski, Jan 2019 paper"

This is a PyTorch implementation of

[Under Construction]

Update April 14, 2019

Began training on Librispeech dev (, see dat/example_train.log


  1. VAE and VQVAE versions of the bottleneck / training objectives
  2. Inference mode

Example training setup


# Get the data
cd $run_dir
tar zxvf dev-clean.tar.gz
$code_dir/scripts/ LibriSpeech/dev-clean > 

# Train
cd $code_dir 
python new -af par/arch.basic.json -tf par/train.basic.json -nb 4 -si 10 \
  -rws 100000 -fpu 1.0 $run_dir/model%.ckpt $run_dir/
You can’t perform that action at this time.