guaini

Results 3 comments of guaini

@mi804 Thanks for your reply. I will try the code later.

@mi804 `selected = attention_map[:, :, 0:512, 1024:5120] selected = selected[:,0,:,:].mean(dim=1)`

How the above code shows the attention map correspond to the word person?