11 Commits

Author	SHA1	Message	Date
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