How to save the attention maps

[rulixiang]

Hey, this is an very interesting work!
I tried to save the cross-attention maps by setting save_cross_attention_maps in config.py to True, but sadly found no attention map was saved. Can you show how to save the attention maps?
Thanks!

[AttendAndExcite]

Hi @rulixiang, thanks for your interest!
This configuration is currently not supported in the code, however, you can use our cross-attention visualization notebook to visualize the cross-attention maps obtained with and without Attend-and-Excite.

[rulixiang]

Thanks, I found it!