Visualize attention weights

[priyanka9991]

Hello,

Thank you for this amazing repository and all the valuable contributions from the team.
I am currently using TITAN as a slide aggregator to obtain slide-level embeddings from patch features. I was wondering if there is a way to access or visualize the attention weights for each patch. This would be extremely helpful for fine-tuning the model on specific downstream tasks.

Thank you,
Priyanka