vibrant-galaxy

Extracting attention maps

3

Hi there, Excellent project! I'm using axial-attention with video (1, 5, 128, 256, 256) and `sum_axial_out=True`, and I wish to visualise the attention maps. Essentially, given my video, and two...