
用于Transformer模型注意力可视化的工具(包括BERT、GPT-2、ALBERT、XLNet、RoBERTa、CTRL等)
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
这是一款专为Transformer模型设计的注意力可视化工具,支持BERT、GPT-2、ALBERT、XLNet、RoBERTa、CTRL等多种预训练模型,帮助用户深入理解模型内部机制。
BertViz 是一个用于在Transformer模型中可视化注意力的工具,支持包括BERT、GPT-2、XLNet、RoBERTa、XLM 和 CTRL在内的所有模型。
它扩展了Llion Jones 的Tensor2Tensor 可视化工具以及HuggingFace 的转换器库。相关的博客文章有《解构BERT,第2部分:可视化注意力的内部工作》,该系列的第一部分并非必要前提条件。此外还有一篇关于“变压器模型中注意力的多尺度可视化的论文”。
全部评论 (0)
还没有任何评论哟~


