14 Commits

Author	SHA1	Message	Date
Fangjun Kuang	1f7832b93c	Fix loading sampler state dict. (#421) ... * Fix loading sampler state dict. * skip scan_pessimistic_batches_for_oom if params.start_batch > 0	2022-08-06 10:00:08 +08:00
Zengwei Yao	8203d10be7	Add stats about duration and padding proportion (#485) ... * add stats about duration and padding proportion * add for utt_duration * add stats for other recipes * add stats for other 2 recipes * modify doc * minor change	2022-07-25 16:40:43 +08:00
Wei Kang	6e609c67a2	Using streaming conformer as transducer encoder (#380) ... * support streaming in conformer * Add more documents * support streaming on pruned_transducer_stateless2; add delay penalty; fixes for decode states * Minor fixes * streaming for pruned_transducer_stateless4 * Fix conv cache error, support async streaming decoding * Fix style * Fix style * Fix style * Add torch.jit.export * mask the initial cache * Cutting off invalid frames of encoder_embed output * fix relative positional encoding in streaming decoding for compution saving * Minor fixes * Minor fixes * Minor fixes * Minor fixes * Minor fixes * Fix jit export for torch 1.6 * Minor fixes for streaming decoding * Minor fixes on decode stream * move model parameters to train.py * make states in forward streaming optional * update pretrain to support streaming model * update results.md * update tensorboard and pre-models * fix typo * Fix tests * remove unused arguments * add streaming decoding ci * Minor fix * Minor fix * disable right context by default	2022-06-28 00:18:54 +08:00
Quandwang	8512aaf585	fix typos (#409)	2022-06-08 20:08:44 +08:00
Fangjun Kuang	f1abce72f8	Use jsonl for CutSet in the LibriSpeech recipe. (#397) ... * Use jsonl for cutsets in the librispeech recipe. * Use lazy cutset for all recipes. * More fixes to use lazy CutSet. * Remove force=True from logging to support Python < 3.8 * Minor fixes. * Fix style issues.	2022-06-06 10:19:16 +08:00
Fangjun Kuang	fce7f3cd9a	Support computing RNN-T loss with torchaudio (#316)	2022-04-19 18:47:13 +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	f721a2fd7a	Minor fixes for logging (#296) ... * Minor fixes for logging * Minor fix	2022-04-10 23:34:18 +08:00
Fangjun Kuang	9a11808ed3	Set the seed for dataloader. (#282) ... Also, suppress torch warnings about division by truncation.	2022-03-31 16:48:46 +08:00
Fangjun Kuang	3ae7265737	More fixes to the checkpoint code. (#266)	2022-03-23 14:37:54 +08:00
Fangjun Kuang	6a091da0b0	Minor fixes for saving checkpoints. (#265) ... * Minor fixes for saving checkpoints. * Fix loading checkpoints saved by previous code.	2022-03-23 12:22:05 +08:00
Fangjun Kuang	ae564f91e6	Periodically saving checkpoint after processing given number of batches (#259) ... * Periodically saving checkpoint after processing given number of batches.	2022-03-20 23:51:33 +08:00
Fangjun Kuang	1c35ae1dba	Reset seed at the beginning of each epoch. (#221) ... * Reset seed at the beginning of each epoch. * Use a different seed for each epoch.	2022-02-21 15:16:39 +08:00
Wei Kang	b702281e90	Use k2 pruned transducer loss to train conformer-transducer model (#194) ... * Using k2 pruned version transducer loss to train model * Fix style * Minor fixes	2022-02-17 13:33:54 +08:00