Skip to content

PyTorch implementation of LF-MMI for End-to-end ASR

Notifications You must be signed in to change notification settings

YiwenShaoStephen/pychain

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch implementation of LF-MMI for End-to-end ASR

End-to-end version of lattice-free MMI (LF-MMI or chain model) implemented in PyTorch.
TODO: regular version of LF-MMI.

What's New:

  • August 2020: GPU computation for graphs in log domain (recommended for numerator graphs)
  • April 2020: Support unequal length sequences within a minibatch
  • April 2020: Examples of using PyChain: Espresso and pychain-example
  • January 2020: GPU computation for both denominator and numerator graphs

Installation and Requirements

First-time Installation (including OpenFST)

pip install kaldi_io
git clone https://github.com/YiwenShaoStephen/pychain.git
cd pychain
make

Update

Whenever you update or modify any none-python codes (e.g. .c or .cu) in pychain, you need to re-compile it by

make pychain

Reference

"PyChain: A Fully Parallelized PyTorch Implementation of LF-MMI for End-to-End ASR", Yiwen Shao, Yiming Wang, Daniel Povey and Sanjeev Khudanpur (pdf)

About

PyTorch implementation of LF-MMI for End-to-end ASR

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published