38 Commits

Author SHA1 Message Date
Fangjun Kuang
897307f445 Add MMI training with word pieces. 2021-08-07 16:41:16 +08:00
Fangjun Kuang
f03c991781 Merge remote-tracking branch 'dan/master' into mmi 2021-08-04 15:00:21 +08:00
Fangjun Kuang
b1b21eb1e4 Fix decoder padding mask. 2021-08-04 14:57:06 +08:00
Fangjun Kuang
5a0b9bcb23
Refactoring (#4)
* Fix an error in TDNN-LSTM training.

* WIP: Refactoring

* Refactor transformer.py

* Remove unused code.

* Minor fixes.
2021-08-04 14:53:02 +08:00
Fangjun Kuang
a6d9b3c9ab Minor fixes. 2021-08-03 22:16:34 +08:00
Fangjun Kuang
2be7a0a555 Remove unused code. 2021-08-03 17:24:06 +08:00
Fangjun Kuang
f6091b10c0 Refactor transformer.py 2021-08-02 23:48:26 +08:00
Fangjun Kuang
1fa30998da WIP: Refactoring 2021-07-31 20:24:47 +08:00
Fangjun Kuang
c72a11ea1f Merge remote-tracking branch 'dan/master' into style-check 2021-07-31 16:49:54 +08:00
Fangjun Kuang
c9222bdb09 Fix an error in TDNN-LSTM training. 2021-07-31 15:55:42 +08:00
Daniel Povey
cf8d76293d
Merge pull request #3 from csukuangfj/style-check
Add CTC training
2021-07-31 15:36:00 +08:00
Fangjun Kuang
398ed80d7a Minor fixes to support DDP training. 2021-07-31 15:26:57 +08:00
Fangjun Kuang
b94d97da37 Disable gradient computation in evaluation mode. 2021-07-29 20:37:31 +08:00
Fangjun Kuang
acc63a9172 WIP: Add BPE training code. 2021-07-29 20:23:52 +08:00
Fangjun Kuang
bd69e4be32 Use attention decoder for rescoring. 2021-07-28 12:22:09 +08:00
Fangjun Kuang
f65854cca5 Add BPE decoding results. 2021-07-27 17:38:47 +08:00
Fangjun Kuang
4ccae509d3 WIP: Begin to add BPE decoding 2021-07-26 20:06:58 +08:00
Fangjun Kuang
d3101fb005 Fix loading checkpoint in DDP training. 2021-07-26 08:08:14 +08:00
Fangjun Kuang
78bb65ed78 Fix an error in DDP training. 2021-07-25 22:33:09 +08:00
Fangjun Kuang
8055bf31a0 Support DDP training. 2021-07-25 21:40:09 +08:00
Fangjun Kuang
4a66712406 Add LM rescoring. 2021-07-25 18:21:26 +08:00
Fangjun Kuang
6f9fe5b906 Refactor decoding code. 2021-07-24 22:23:50 +08:00
Fangjun Kuang
00f8371f37 begin to add LM rescoring. 2021-07-24 18:24:04 +08:00
Fangjun Kuang
a9095925ba Fix CI test errors. 2021-07-24 18:13:03 +08:00
Fangjun Kuang
54436182a4 Fix CI. 2021-07-24 18:05:19 +08:00
Fangjun Kuang
ee83a3e67c Fix CI dependencies installation. 2021-07-24 17:55:45 +08:00
Fangjun Kuang
2e33e24348 Add CI test. 2021-07-24 17:47:41 +08:00
Fangjun Kuang
f3542c7793 Add CTC training. 2021-07-24 17:13:20 +08:00
Fangjun Kuang
a01d08f73c Add self-loops to propagate disambiguation symbols. 2021-07-21 13:12:20 +08:00
Fangjun Kuang
8a72901f3a Minor fixes. 2021-07-20 19:54:12 +08:00
Fangjun Kuang
d5e0408698 Add prepare_lang.py based on prepare_lang.sh 2021-07-20 19:41:21 +08:00
Fangjun Kuang
e005ea062c Minor fixes after review. 2021-07-20 10:02:20 +08:00
Fangjun Kuang
f25eedf2d4 Fixes after review. 2021-07-20 00:14:24 +08:00
Fangjun Kuang
0b19aa09c1 Compute features of librispeech and musan. 2021-07-19 23:35:32 +08:00
Fangjun Kuang
40eed74460 Download LM for LibriSpeech. 2021-07-15 21:09:14 +08:00
Fangjun Kuang
d146a4e799 Remove mypy. 2021-07-15 19:52:01 +08:00
Fangjun Kuang
71c4e29ad5 Add style check tools. 2021-07-15 17:36:48 +08:00
Fangjun Kuang
0d16431766 First commit. 2021-07-15 17:35:54 +08:00