fairseq/modules/relative_multihead_attention.py · 557024664b6cc3d1a65ea08ce53e5ae695e8a661 · xuchen / Fairseq-S2T · GitLab

Blame History Permalink

fix the bugs and use float32 for softmax · bab6c520
xuchen committed Feb 24, 2022

bab6c520

relative_multihead_attention.py 14 KB