lucidrains / DALLE-pytorch

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Visualizing Attention Map

EmaadKhwaja opened this issue · 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.