search

AlexHex7 / Non-local_pytorch

Implementation of Non-local Block.
Apache License 2.0
1.57k stars 277 forks source link

Visualization #30

Open ShairozS opened 5 years ago

ShairozS commented 5 years ago

Awesome library, I was able to convert it to train against my own dataset after making some modifications. Are there any plans to include (or do you have something personal written up) for doing attention masks or any other visualizations (such as the one in https://arxiv.org/pdf/1805.08318.pdf)? Trying to understand the non-local dependencies the model is forming

AlexHex7 commented 4 years ago

@ShairozS Hi, I write a demo to visualize Non-local Attention Map. I hope it can helps you.