About the figure5

[Duranin]

Hello, how did you draw a heatmap? The effect I drew using the Gradcam method is much worse than the one in your paper. Would you like to make this part of the code public?

[goutamyg]

Hi, thank you for your interest in our work. The code to visualize the attention heatmaps is already present in the codebase here.

The function that creates the visualization is defined here