8 Commits

Author SHA1 Message Date
zr_jin
17ad6c2959 Update alignment_attention_module.py 2023-07-23 20:24:58 +08:00
zr_jin
7e5c7e6f77 fixes on feat dim 2023-07-23 20:19:19 +08:00
zr_jin
3bd2e8e6cc fixed feat dim 2023-07-23 18:20:55 +08:00
JinZr
bb8d016722 minor fixes 2023-07-23 18:15:47 +08:00
zr_jin
7824789a52 fixed 2023-07-23 18:13:30 +08:00
zr_jin
ab9affd3e5 updated model.py to incorporate AlignmentAttnModule
output of the attn module will replace the am_pruned as the input of the joiner model
2023-07-23 18:10:39 +08:00
zr_jin
706cbae5d0 init commit for the alignment attn module 2023-07-23 18:07:32 +08:00
zr_jin
fb410cc46f init commit 2023-07-23 13:15:22 +08:00