Open ShirGrinblat opened 1 year ago
Hey, I'm looking for a function that visualize the attention of every VIT layer (like Fig 4 in the article) do you provide it?
Thanks
Have you solved it? I also want a method to output the attention of each layer.
Hey, I'm looking for a function that visualize the attention of every VIT layer (like Fig 4 in the article) do you provide it?
Thanks