Hello, I wonder if anyone knows how to extract attentions from the last layer of UNI? I want to plot attention maps for each head. Thanks in advance!
In general, pytorch's hook is your friend.
· Sign up or log in to comment
