how to generate the visualization of the attention map between the CLS token(Fig 4 in the paper). Can you provide the source code? Thanks a lot.
how to generate the visualization of the attention map between the CLS token(Fig 4 in the paper). Can you provide the source code? Thanks a lot.