/bertviz

Tool for visualizing attention in the Transformer model (BERT, GPT-2, Albert, XLNet, RoBERTa, CTRL, etc.)

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers