Visualize attention weights

[fbrad]

Hi! I was wondering if you guys plan on adding support for easy retrieval of the attention weights during decoding? This would be really helpful for qualitative analysis.

Thanks!

[srush]

Yes, we are working on this. We will try to get it in at least a branch soon-ish.

[sebastianGehrmann]

Just pushed a file to the branch attention_retrieval

[yoonkim]

thanks sebastian! closing