pulling attention maps from UNI

by Siyuan726 - opened Mar 24, 2024

Mar 24, 2024

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!

Apr 8, 2024

In general, pytorch's hook is your friend.

GCabas

Jan 3

Did you resolved your issue?

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment