Open Hezey opened 1 year ago
@HaoZhang534 @SlongLiu @FengLi-ust @IDEACVR Excuse me...Again, I have a question. How to Visualize the attentions of the model? 1.how to visualize encoder-decoder multi-head attention weights? 2.how to visualize encoder self-attention weights Thanks very much!
You need to get the attention map you want and normalize it to 0~255 and then you can save it as an image.
@HaoZhang534 @SlongLiu @FengLi-ust @IDEACVR Excuse me...Again, I have a question. How to Visualize the attentions of the model? 1.how to visualize encoder-decoder multi-head attention weights? 2.how to visualize encoder self-attention weights Thanks very much!