transformer-pytorch icon indicating copy to clipboard operation
transformer-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon dreamgonfly

how can we get attention weights for visualization

Open liperrino opened this issue 4 years ago • 0 comments

Hi. Please i am very interested in your code and i would like to know how to extract weights for attention visualization. If possible how to do visualization. Your code is great and thanks a lot.

liperrino avatar Apr 05 '20 01:04 liperrino