Visualizer 是一个辅助深度学习模型中 Attention 模块可视化的小工具，主要功能是帮助取出嵌套在模型深处的 Attention Map。

具有以下特点：

  • 精准直接，你可以取出任何变量名的模型中间结果；

  • 快捷方便，一个操作，就可以同时取出 Transformer 类模型中的所有 attention map；

  • 非侵入式，你无须修改函数内的任何一行代码；

  • 训练-测试一致，可视化完成后，训练时无须再将代码改回来。

项目主页：
https://github.com/luo3300612/Visualizer

内容中包含的图片若涉及版权问题，请及时与我们联系删除