fairseq/modules/espnet_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

espnet_multihead_attention.py 10.8 KB