tomename

How do you visualize attention weights?

1

Thanks for sharing the great work. I would like to reproduce the paper's visualization results for the decoder self-attention weights in Figure 1 and the encoder self-attention weights in Figure...

Weight-based alignment function "get_acts_wassersteinized_layers_modularized()" does not implement to handle “handle-skips” and the following error occurs when used on ResNet-18. ``` RuntimeError: Expected size for first two dimensions of batch2 tensor...