lucidrains/DALLE-pytorch

Visualizing Attention Map

EmaadKhwaja opened this issue · 1 comments

Has anyone had success visualizing attention weights for images and text tokens. I'm really interested in seeing why the model is selecting tokens.

Hi, have you got the visualized results? I'm also interested in this issue.