search

mt-upc / transformer-contributions-nmt

Apache License 2.0
13 stars 7 forks source link

save_activation #3

Open comea23 opened 4 months ago

comea23 commented 4 months ago

The save_activation in m2m demo is unable to save the inputs and outputs of q_proj, k_proj, and v_proj within the self_atten submodule.

comea23 commented 4 months ago

solved