18 Commits

Author SHA1 Message Date
JinZr
634931cb61 minor updates 2023-07-27 17:52:49 +08:00
JinZr
d70f6e21f2 updated code for decoding 2023-07-26 19:01:53 +08:00
jinzr
79b8e60f93 minor fixes 2023-07-25 23:26:04 +08:00
JinZr
755430c29e fixes 2023-07-25 19:56:58 +08:00
JinZr
3b4fa4863f minor fixes 2023-07-25 16:05:57 +08:00
JinZr
90cb518398 update 2023-07-25 16:02:48 +08:00
JinZr
49e9d15733 issues fixed 2023-07-24 17:30:24 +08:00
zr_jin
739e2a22c6 moved an operand to the joiner model 2023-07-23 22:30:28 +08:00
zr_jin
056efeef30 load attn param 2023-07-23 22:18:25 +08:00
zr_jin
8bc0956503 Create alignment_attention_module_debug.py
logging tensor dims for debugging
2023-07-23 21:15:02 +08:00
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