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 |
|