15 Commits

Author SHA1 Message Date
yaozengwei
61ecd3764d minor fix of doc of pos_emb 2022-05-10 12:15:14 +08:00
yaozengwei
5de9d0a19a add tanh_on_mem option 2022-05-10 11:48:22 +08:00
yaozengwei
e3a29b17f3 minor refactor of emformer codes 2022-05-10 11:15:44 +08:00
yaozengwei
50fe100f50 support position encoding for emformer 2022-05-04 20:11:50 +08:00
yaozengwei
24db3a1934 update emformer_pruned_transducer_stateless/emformer.py 2022-04-20 14:21:45 +08:00
yaozengwei
328ad280a4 Support state stacking and unstacking operations for emformer_pruned_transducer_stateless/emformer.py 2022-04-19 17:58:51 +08:00
yaozengwei
524f3aa015 update test functions for emformer. 2022-04-14 16:41:52 +08:00
yaozengwei
4130892971 delete duplicated dropout in emformer attention and update emformer test codes. 2022-04-13 23:46:42 +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
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
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