search

juhongm999 / pervit

Official PyTorch Implementation of Peripheral Vision Transformer, NeurIPS 2022
Apache License 2.0
40 stars 2 forks source link

Figure 6: Visualization of learned attentions Φp and Φa for l ∈ {3, 4, 8}. #2

Open Q418 opened 1 year ago

Q418 commented 1 year ago

Hello, can the attention visualization code of Figure 6 in the paper be provided? Thank you so much.

juhongm999 commented 1 year ago

You can try adding --visualize option during testing.

Q418 commented 1 year ago

Sorry, I tried that, adding the visualization during testing only gets figure 4（Figure 4: Learned attentions Φp of PerViT-T）.