9 Commits

Author	SHA1	Message	Date
yaozengwei	1f6c822dc0	add @torch.jit.export for init_states function	2022-07-06 17:46:36 +08:00
yaozengwei	dbea9a9970	Merge remote-tracking branch 'k2-fsa/master' into emformer_conv_simplify_new	2022-07-05 15:47:41 +08:00
yaozengwei	10662c5c38	fix bug about memory mask when memory_size==0	2022-07-04 17:05:47 +08:00
yaozengwei	9c37c16326	modify .flake8	2022-06-22 12:28:26 +08:00
yaozengwei	42e3e883fd	refactor init states for stream	2022-06-21 22:07:36 +08:00
yaozengwei	5b19011edb	fix doc in conv_emformer_transducer_stateless/emformer.py	2022-06-21 20:52:42 +08:00
Zengwei Yao	a42d96dfe0	Fix warmup (#435) ... * fix warmup when scan_pessimistic_batches_for_oom * delete comments	2022-06-20 13:40:01 +08:00
yaozengwei	5d877ef25e	change tail padding length from right_context_length to chunk_length	2022-06-17 18:54:55 +08:00
Zengwei Yao	53f38c01d2	Emformer with conv module and scaling mechanism (#389) ... * copy files from existing branch * add rule in .flake8 * monir style fix * fix typos * add tail padding * refactor, use fixed-length cache for batch decoding * copy from streaming branch * copy from streaming branch * modify emformer states stack and unstack, streaming decoding, to be continued * refactor Stream class * remane streaming_feature_extractor.py * refactor streaming decoding * test states stack and unstack * fix bugs, no grad, and num_proccessed_frames * add modify_beam_search, fast_beam_search * support torch.jit.export * use torch.div * copy from pruned_transducer_stateless4 * modify export.py * add author info * delete other test functions * minor fix * modify doc * fix style * minor fix doc * minor fix * minor fix doc * update RESULTS.md * fix typo * add info * fix typo * fix doc * add test function for conv module, and minor fix. * add copyright info * minor change of test_emformer.py * fix doc of stack and unstack, test case with batch_size=1 * update README.md	2022-06-13 15:09:17 +08:00