427 Commits

Author	SHA1	Message	Date
Guo Liyong	78418ac37c	fix comments	2022-04-13 13:09:24 +08:00
yaozengwei	c2808f8541	Support cache of left context for causal convolution.	2022-04-12 20:13:51 +08:00
Daniel Povey	2a854f5607	Merge pull request #309 from danpovey/update_results ... Update results; will further update this before merge	2022-04-12 12:22:48 +08:00
Daniel Povey	9ed7a169e1	Add one more epoch of full expt	2022-04-12 12:20:10 +08:00
Daniel Povey	d0a53aad48	Fix tensorboard log location	2022-04-12 11:51:15 +08:00
Daniel Povey	65818d16de	Add more results	2022-04-12 11:48:16 +08:00
Fangjun Kuang	bdeff338c2	Fix CI errors. (#310)	2022-04-12 09:09:56 +08:00
Mingshuang Luo	118e195004	Update results for tedlium3 pruned RNN-T (#307) ... * Update README.md	2022-04-11 22:19:26 +08:00
Mingshuang Luo	93c60a9d30	Code style check for librispeech pruned transducer stateless2 (#308)	2022-04-11 22:15:18 +08:00
Daniel Povey	ead822477c	Fix rebase	2022-04-11 21:01:13 +08:00
Daniel Povey	e8eb0b94d9	Updating RESULTS.md; fix in beam_search.py	2022-04-11 21:00:11 +08:00
pkufool	a92133ef96	Minor fixes	2022-04-11 20:58:47 +08:00
pkufool	ddd8f9e15e	Minor fixes	2022-04-11 20:58:43 +08:00
pkufool	cc0d4ffa4f	Add mix precision support	2022-04-11 20:58:02 +08:00
Mingshuang Luo	8cb727e24a	Tedlium3 pruned transducer stateless (#261) ... * update tedlium3-pruned-transducer-stateless-codes * update README.md * update README.md * add fast beam search for decoding * do a change for RESULTS.md * do a change for RESULTS.md * do a fix * do some changes for pruned RNN-T	2022-04-11 17:08:53 +08:00
Wei Kang	7012fd65b5	Support mix precision training on the reworked model (#305) ... * Add mix precision support * Minor fixes * Minor fixes * Minor fixes	2022-04-11 16:49:54 +08:00
Daniel Povey	03c7c2613d	Fix docs in optim.py	2022-04-11 15:13:42 +08:00
Daniel Povey	6eb6d9b4cd	Merge pull request #288 from danpovey/reworked_model ... Reworked model	2022-04-11 15:03:08 +08:00
Daniel Povey	5078332088	Fix adding learning rate to tensorboard	2022-04-11 14:58:15 +08:00
yaozengwei	651745b220	minor fix doc in emformer.py	2022-04-11 12:42:47 +08:00
yaozengwei	6c1f9b5181	Add wenet ref in ConvolutionModule class.	2022-04-11 12:38:45 +08:00
Daniel Povey	d5f9d49e53	Modify beam search to be efficient with current joienr	2022-04-11 12:35:29 +08:00
yaozengwei	1d74c5e596	Support causal convolution in emformer encoder layer.	2022-04-11 12:28:15 +08:00
Daniel Povey	46d52dda10	Fix dir names	2022-04-11 12:03:41 +08:00
Wei Kang	f721a2fd7a	Minor fixes for logging (#296) ... * Minor fixes for logging * Minor fix	2022-04-10 23:34:18 +08:00
Fangjun Kuang	7f73043219	Merge remote-tracking branch 'dan/master' into streaming	2022-04-10 23:25:56 +08:00
yaozengwei	a24eef8096	update conv_emformer_transducer/emformer.py.	2022-04-10 20:29:22 +08:00
yaozengwei	8129470586	first upload the conv_emformer_transducer recipe, integrating convolution module into emformer layers.	2022-04-10 20:24:20 +08:00
Daniel Povey	962cf868c9	Fix import	2022-04-10 15:31:46 +08:00
Daniel Povey	d1e4ae788d	Refactor how learning rate is set.	2022-04-10 15:25:27 +08:00
Daniel Povey	82d58629ea	Implement 2p version of learning rate schedule.	2022-04-10 13:50:31 +08:00
Daniel Povey	da50525ca5	Make lrate rule more symmetric	2022-04-10 13:25:40 +08:00
Daniel Povey	4d41ee0caa	Implement 2o schedule	2022-04-09 18:37:03 +08:00
Daniel Povey	db72aee1f0	Set 2n rule..	2022-04-09 18:15:56 +08:00
yaozengwei	3e131891a2	update test_emformer.py	2022-04-08 20:43:54 +08:00
yaozengwei	d58002c414	update emformer.py	2022-04-08 20:31:32 +08:00
Daniel Povey	0f8ee68af2	Fix bug	2022-04-08 16:53:42 +08:00
Daniel Povey	f587cd527d	Change exponential part of lrate to be epoch based	2022-04-08 16:24:21 +08:00
Daniel Povey	6ee32cf7af	Set new scheduler	2022-04-08 16:10:06 +08:00
Fangjun Kuang	78b8792d1d	Fix potential bugs in PyTorch that exist in label_smoothing. (#300)	2022-04-08 13:41:33 +08:00
yaozengwei	2d1b90f758	update the docs of Emformer class in emformer.py	2022-04-08 10:59:39 +08:00
yaozengwei	374eacdd5c	First upload emformer_pruned_transducer_stateless recipe, refator emformer codes from torchaudio.	2022-04-07 21:32:59 +08:00
Daniel Povey	61486a0f76	Remove initial_speed	2022-04-06 13:17:26 +08:00
Daniel Povey	a41e93437c	Change some defaults in LR-setting rule.	2022-04-06 12:36:58 +08:00
yaozengwei	9423b3899f	Update emformer_pruned_transducer_stateless/emformer.py and upload emformer_pruned_transducer_stateless/test_emformer.py.	2022-04-05 22:28:46 +08:00
yaozengwei	fe43c1349e	First upload emformer_pruned_transducer_stateless/emformer.py, modified from torchaudio.	2022-04-05 22:28:46 +08:00
yaozengwei	b0bce20e21	Modify subsampling.py to make T'=T//4 strictly	2022-04-05 22:28:46 +08:00
Daniel Povey	2545237eb3	Changing initial_speed from 0.25 to 01	2022-04-05 18:00:54 +08:00
Daniel Povey	25724b5ce9	Bug-fix RE sign of target_rms	2022-04-05 13:49:35 +08:00
Daniel Povey	d1a669162c	Fix bug in lambda	2022-04-05 13:31:52 +08:00