search

tensorflow / tensor2tensor

Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.
Apache License 2.0
15.5k stars 3.49k forks source link

attention visulization #1341

Open HqWei opened 5 years ago

HqWei commented 5 years ago

Do you mind to provide the code or details about how to implement the visualization of the attention map?

artitw commented 5 years ago

Try this example notebook. I can also provide more examples if necessary. https://github.com/tensorflow/tensor2tensor/blob/master/tensor2tensor/visualization/TransformerVisualization.ipynb